如何将YOLOv3自定义训练的* .pth文件更改为darknet格式或caffemodel?

时间:2019-12-03 04:51:31

标签: python pytorch yolo opencv-python

如何更改自己训练有素的检查点文件的格式?这是经过pytorch训练的yolov3。我在Darknet类中找到了'save_darknet_weights'方法,但这没有帮助。这是我将* .pth更改为暗网重量格式的代码;

device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
model = Darknet('yolov3-custom.cfg').to(device)
model.load_state_dict(torch.load('checkpoint.pth', map_location=torch.device('cpu')))

Darknet.save_darknet_weights(model, 'converted.weights', cutoff=-1)

已在我的项目文件夹中创建了'converted.weights'文件,但使用'cv.dnn.readNetFromDarknet'进行的推断未检测到任何东西。有什么问题吗?

0 个答案:

没有答案