要调用best.pt的项目路径:
当前项目的路径如下:
我要在yolov5-D435i-main项目中去调用yolov5-6.0项目中训练好的best.pt文件,然后得把yolov5-6.0项中你要用到的best.pt文件放在yolov5-D435i-main项目中。
model = torch.hub.load('E:/yolo/yolov5-6.0', 'custom', path=r'best.pt', source='local') # local repo
model.conf = 0.8 # set confidence threshold
如此,成功!!!