bsrgan 0.0.4

Packaged version of the BSRGAN repository

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')
model.save = True

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}
}