yolov5主目录下找到detect.py文件,打开该文件。
往下面翻,主要对是weights和source处修改:
加载模型:
找到这行并修改,’default‘选择我们刚刚训练好的模型(best.pt)。(注意路径)
parser.add_argument('--weights', nargs='+', type=str, default='runs/train/exp13/weights/best.pt', help='model.pt path(s)')
选择图像进行检测:(图片视频default修改路径就行如'test1.MP4',摄像头default为0)(注意路径)
parser.add_argument('--source', type=str, default='TEST1.mp4', help='source') #file/dir/URL/glob/screen/0(webcam)
运行程序,会提示生成的检测结果存在哪里,进入文件夹后可以查看。如下图。