chenjunson

> 是可以的，需要你改一下那个model.py内部加载weights的函数，把遇到你新加模块那边的替换就可以了，你可以参照我用80分类预训练模型给不同分类模型做预训练的写法，就是-pt那部分 我在原来三层yolo检测层的基础上又添加了一个4倍降采样的yolo检测层，现在总共四个yolo检测层，结果在调用预训练模型时出现了这个报错：chenjunsong@chenjunsong-GJ5CN64:~/U-YOLOv3$ python3 train.py --data data/obj.data --batch-size 4 --cfg cfg/yolov3-mobilenet/yolov3-mobilenet-yu-1.cfg --weights weights/yolov3-mobilenet.weights -pt Apex recommended for faster mixed precision training: https://github.com/NVIDIA/apex Namespace(KDstr=-1, adam=False, batch_size=4, bucket='', cache_images=False, cfg='./cfg/yolov3-mobilenet/yolov3-mobilenet-yu-1.cfg', data='data/obj.data', device='',...