先说方法,关闭加载预训练,在train.py进行如下修改
def create_model(input_shape, anchors, num_classes, load_pretrained=False, freeze_body=2,
weights_path='model_data/yolo_weights.h5'):
def create_tiny_model(input_shape, anchors, num_classes, load_pretrained=False, freeze_body=2,
weights_path='model_data/tiny_yolo_weights.h5'):
也就是将原来的 load_pretrained=True 修改为 Fasle。
至于原因,一开始我把网上的方法试了一遍都没用,然后我看了一下我的数据集和voc数据集,感觉我的数据集和voc的内容相差特别大,就盲猜关掉预训练,结果就可以了。