如何更改自己训练有素的检查点文件的格式?这是经过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'进行的推断未检测到任何东西。有什么问题吗?