YOLOv6开源链接:https://github.com/meituan/YOLOv6
下载权重模型,本文使用的是YOLOv6-S
1、安装需要的包
打开工程后进入终端输入
pip install requirements.txt
2、修改infer.py
找到文件夹tools->infer.py
默认的路径需要改一下,否则会报如下错误
这里我直接修改成绝对路径
modelpath-----模型存放路径
imgpath-----图片存放路径
yamlpath-----yaml文件存放路径
修改完成后运行,出现如下错误:
3、修改inferer.py
此时需要再找到文件夹yolov6->core->inferer.py
找到文件254行,在路径加个点
到此就完成配置了,然后运行infer.py文件
运行的结果保存在下图位置: