Views
No views yet
pip install yolov6detect1from yolov6 import YOLOV6
2
3model = YOLOV6(weights='kadirnar/yolov6s-v2.0', device='cuda:0', hf_model=True)
4
5model.classes = None
6model.conf = 0.25
7model.iou = 0.45
8model.show = False
9model.save = True
10
11pred = model.predict(source='data/images',yaml='data/coco.yaml', img_size=640)@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}
}