pytorch 如何在真实的检测应用中加载yolov7模型?

jdzmm42g  于 2022-12-23  发布在  其他
关注(0)|答案(1)|浏览(252)

我使用yolov5对我的自定义对象进行对象检测,并用于加载模型,如下所示:

model = torch.hub.load("/Users/macbook/Documents/Objection-Detection/yolov5", "custom", source="local",
                       path=os.path.join("/Users/macbook/Documents/Objection-Detection/", model_name),
                       force_reload=True)

我在yolov7 google colab中训练了我的自定义对象,在下载了模型之后,我想把它加载到我的PyCharm应用程序中进行检测。那么有人能告诉我如何像yolov5那样做吗?
附言:别提我的英语错误,我是个法国人,还在学英语
我所要求的是,如果有一种方法来加载yolov7模型这样:

model = torch.hub.load("ultralytics/yolov5", "yolov5s", pretrained=True)
a64a0gku

a64a0gku1#

torch.hub中没有Yolov7型号。查看:
https://pytorch.org/hub/
然后搜索Yolo。你会找到的唯一型号是Yolov5和YoloP

相关问题