BSRGAN-Pip: Packaged version of the BSRGAN repository
Overview
This repo is a packaged version of the BSRGAN model.
Installation
pip install bsrgan
BSRGAN Usage
from bsrgan import BSRGAN
model = BSRGAN(weights='kadirnar/bsrgan', device='cuda:0')
pred = model.predict(img_path='data/image/test.png')
Citation
@article{li2022yolov6,
title={YOLOv6: A single-stage object detection framework for industrial applications},
author={Li, Chuyi and Li, Lulu and Jiang, Hongliang and Weng, Kaiheng and Geng, Yifei and Li, Liang and Ke, Zaidan and Li, Qingyuan and Cheng, Meng and Nie, Weiqiang and others},
journal={arXiv preprint arXiv:2209.02976},
year={2022}
}