我使用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)
1条答案
按热度按时间a64a0gku1#
torch.hub
中没有Yolov7型号。查看:https://pytorch.org/hub/
然后搜索Yolo。你会找到的唯一型号是Yolov5和YoloP