yolov6 学习初篇

1、下载官方code

GitHub - meituan/YOLOv6: YOLOv6: a single-stage object detection framework dedicated to industrial applications.

2、下载预训练模型

在readme中有下载地址:

[**YOLOv6-n**](https://github.com/meituan/YOLOv6/releases/download/0.1.0/yolov6n.pt)

[**YOLOv6-tiny**](https://github.com/meituan/YOLOv6/releases/download/0.1.0/yolov6t.pt)

[**YOLOv6-s**](https://github.com/meituan/YOLOv6/releases/download/0.1.0/yolov6s.pt) 

将模型文件放在新建的weights文件夹下

3、infer推理

执行指令

python tools/infer.py --weights weights/yolov6t.pt --source data/images/image1.jpg --device 0

输出结果

猜你喜欢

转载自blog.csdn.net/m0_37264397/article/details/125628566