系统环境:Win10+Python3.6.5+YOLOV3
错误提示:Sizes of tensors must match except in dimension 1. Got 85 and 6 in dimension 2
该错误主要是由于cfg文件内容配置错误
cfg文件当中的classes的默认值为80,filters初始值为255。
当中的classes数目需要根据自己训练模型进行修改,filters的值根据classes的值和num的值计算的来
如果识别种类为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
版权声明:本文为CSDN博主「SYC_MORE」的原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接及本声明。
原文链接:https://blog.csdn.net/qq_40902709/article/details/111358121
暂无评论