Yolov5在Tensorflow自定义重量负载中的应用

Question

是否有一种方法可以在python中像这样加载yolov5：

model = xyz.load("yolov5", weight_path)

因此，在模型变量中，我有了整个yolov5的体系结构

Answer 1

使用torch.hub.load：

model = torch.hub.load(
            path-to-hubconf-file(eg. yolo_dir),
            'custom',
            path=(path-to-weight-file(eg. yolov5/weights/<model_name>.pt),
            source='local',
            force_reload=True,
        )

然后我们可以使用：

model.conf = 0.45