Packaged version of the BSRGAN repository
Project description
Yolov6-Pip: Packaged version of the Yolov6 repository
Overview
This repo is a packaged version of the Yolov6 model.
Installation
pip install yolov6detect
Yolov6 Inference
from yolov6 import YOLOV6
model = YOLOV6(weights='yolov6s.pt', device='cuda:0')
model.classes = None
model.conf_thres = 0.25
model.iou_thresh = 0.45
model.view_img = False
model.save_img = True
pred = model.predict(source='data/images',yaml='data/coco.yaml', img_size=640)
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}
}
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
bsrgan-0.0.1.tar.gz
(27.1 kB
view hashes)
