yolov5测试步骤

最新推荐文章于 2024-05-02 02:18:57 发布

小云讲运筹

最新推荐文章于 2024-05-02 02:18:57 发布

阅读量280

点赞数

分类专栏：深度学习文章标签： YOLO python 人工智能

本文链接：https://blog.csdn.net/weixin_48888528/article/details/128835287

版权

深度学习专栏收录该内容

14 篇文章 0 订阅

订阅专栏

1、
last加载last.pt，当作初始化。
在这里插入图片描述
detect.py 有模型权重文件后才能进行测试。
配置文件：

执行脚本detect.py 修改下面的的参数并运行：

weights：需要默认最好的权重，才能预测出结果。

parser.add_argument('--weights', nargs='+', type=str, default='/home/menghan/YOLO5/YOLO5/yolov5-master/runs/exp49/weights/best.pt', help='model.pt path(s)')
parser.add_argument('--source', type=str, default='../NEU-DET/valid/images/inclusion_5.jpg', help='source')  # file/folder, 0 for webcam
parser.add_argument('--output', type=str, default='inference/output', help='output folder')  # output folder