在cfg/YOLO3.cfg的文件上改,生成自己的cfg/yolo3-face.cfg
1 [net] 2 # Testing 3 # batch=1 4 # subdivisions=1 5 # Training 6 batch=64 7 subdivisions=16 8 width=416 9 height=416 10 channels=3 11 momentum=0.9 12 decay=0.0005 13 angle=0 14 saturation = 1.5 15 exposure = 1.5 16 hue=.1
其中:
batch=64 每batch个样本更新一次参数。
subdivisions=16 如果内存不够大,将batch分割为subdivisions个子batch,每个子batch的大小为batch/subdivisions。
训练的话把上面注释掉,测试就把训练部分的注释掉