系统环境:Win10+Python3.6.5+YOLOV3
1.错误提示
Sizes of tensors must match except in dimension 1. Got 85 and 6 in dimension 2
2.错误原因
该错误主要是由于cfg文件内容配置错误
cfg文件当中的classes的默认值为80,filters初始值为255。
当中的classes数目需要根据自己训练模型进行修改,filters的值根据classes的值和num的值计算的来
3.解决办法
如果识别种类为1时:classes=1,num=6,filters=18。
[convolutional]
size=1
stride=1
pad=1
filters=18
activation=linear
[yolo]
mask = 0,1,2
anchors = 10,14, 23,27, 37,58, 81,82, 135,169, 344,319
classes=1
num=6
jitter=.3
ignore_thresh = .7
truth_thresh = 1
random=1