| license: gpl-3.0 | |
| inference: false | |
| tags: | |
| - object-detection | |
| - computer-vision | |
| - yolov6 | |
| - pypi | |
| datasets: | |
| - detection-datasets/coco | |
| ### Installation | |
| ``` | |
| pip install yolov6detect | |
| ``` | |
| ### Yolov6 Inference | |
| ```python | |
| 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) | |
| ``` |