训练结束后,在runs/train/exp/weights目录下会产生两个权重文件,一个是训练效果最好的权重文件,一个是训练最后一轮的权重文件
1、图片测试步骤
1.将best.pt权重文件复制粘贴在主目录的weights文件夹下(方便后续操作)
2.将想进行测试的图片放在主目录的data文件夹下,如图所示:
3.打开defect.py,更改以下部分:
将"–weight"改为best.pt权重文件的地址(在weights文件夹中)
将"–source"改为要测试的图片地址(在data文件夹中)
parser.add_argument('--weights', nargs='+', type=str, default='weights/best.pt', help='model.pt path(s)')
parser.add_argument('--source', type=str, default='data/2.jpg', help='source') # file/folder, 0 for webcam
如图所示:
4.运行defect.py,即可得到结果:
由运行栏可知,输出结果保存在了runs\detect\exp3目录下,找到输出结果打开如下图所示:
(不同电脑地址可能不同)
5.如果识别视频,在将"–source"改为要测试的视频地址即可(注意加.mp4后缀名)
如果识别摄像头,在将"–source"的值置0即可,即:
parser.add_argument('--source', type=str, default='0', help='source') # file/folder, 0 for webcam