(资料图片)

相关源码贴图如下：

importtorchimportcv2ascv#loadimagedataimg="data/images/zidane.jpg"#加载本地预训练模型model=torch.hub.load(repo_or_dir="D:/python/yolov5-7.0-attention/",model="yolov5s",source="local")#loadimage#img="D:/bird_test/test004.png"#加载本地自定义模型#model=torch.hub.load("D:/python/yolov5-7.0-attention/","custom",path="uav_bird.pt",source="local")results=model(img)#显示frame=results.render()[0]bgr=cv.cvtColor(frame,cv.COLOR_RGB2BGR)cv.imshow("PytorchHub+YOLOv5CustomObjectDetection",bgr)cv.waitKey(0)

扫码查看OpenCV+OpenVIO+Pytorch系统化学习路线图