yolov4-keras icon indicating copy to clipboard operation
yolov4-keras copied to clipboard

Published 20 hours ago • verified publisher icon bubbliiiing

训练自己的数据集shape不匹配问题

Open Seibelgys opened this issue 2 years ago • 1 comments

博主您好，我学习了您在github上的很多项目，比如yolov4，我想要研究缺陷检测，于是自己给数据集打了标签，一共有6个类别，在yolov3的相关代码文件夹里进行了替换，并且voc_classes.txt也换成了自己的类别，但是在运行predict或者get_map的时候就会出现维度shape不匹配的错误，是因为coco_classes.txt和代码里coco的类别默认为80的原因吗，我把model_path和class_path的路径也进行了更换，但是还是会提示shape不匹配，在yolo.py里把num_classes设置成等于6也还是shape不匹配，想请问博主这种情况怎么办呢

2 3 1 4

Seibelgys avatar Mar 28 '23 12:03 Seibelgys

预测的文件看看

bubbliiiing avatar Apr 07 '23 17:04 bubbliiiing