如果修改了yolov3spp的网络结构,想要在训练时使用yolov3-spp-ultralytics-512.pt预训练模型,而不想去从头开始用coco数据集去重新训练一个预训练权重,可以读取跟修改后的部分权重进行匹配,其他未匹配成功的层权重重新训练吗?因为使用您的代码之后,cfg和weights没匹配成功报如下错误,
报错内容
The above exception was the direct cause of the following exception:
Traceback (most recent call last):
File "train.py", line 290, in <module>
train(hyp)
File "train.py", line 106, in train
raise KeyError(s) from e
KeyError: "weights/yolov3-spp-ultralytics-512.pt is not compatible with cfg/my_yolov3_spp3.cfg. Specify --weights '' or specify a --cfg compatible with weights/yolov3-spp-ultralytics-512.pt. See https://github.com/ultralytics/yolov3/issues/657"
但是如果使用-- weights '' ,就是从头训练了,我的GPU没办法支持如此高额成本的运算
我想要达到修改了yolov3spp的网络结构,想要在训练时使用yolov3-spp-ultralytics-512.pt预训练模型。