ハク 2021-12-22 14:48
浏览 638
已结题

请问一下,如果修改了网络结构,想要在训练时使用预训练模型,该怎么做呀?

如果修改了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预训练模型。
  • 写回答

0条回答 默认 最新

    报告相同问题?

    问题事件

    • 系统已结题 12月30日
    • 创建了问题 12月22日

    悬赏问题

    • ¥20 ARKts悬浮窗和快捷方式问题
    • ¥15 comparecluster没有办法kegg
    • ¥15 远程访问linux主机超时
    • ¥15 odoo17存货管理优势于中国国内该行业传统ERP或MES的详细解读和举例
    • ¥15 CPU卡指令整合指令数据都在图片上
    • ¥15 odoo17处理受托加工产品
    • ¥15 如何用MATLAB编码图三的积分
    • ¥15 圆孔衍射光强随孔径变化
    • ¥15 MacBook pro m3max上用vscode运行c语言没有反应
    • ¥15 ESP-PROG配置错误,ALL ONES