keras-yolov3在训练自定义图片集的时候,必须修改yolo3.cfg配置文件的相关参数。
主要修改三个yolo部分,每一处都要修改三个地方。
filters:3*(5+len(classes));
classes: len(classes) = 16,这里共有十六中缺陷分类
random:改为0
random=0代表了多尺度训练不开启,不开启对显存的要求会更低,这样显存不够的显卡也可以训练,如果random=1那么多尺度训练开启,在训练中图片会随机增强放大等等,对显存要求更高,这里设为0即可。