Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

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

Open
Seibelgys opened this issue Mar 27, 2023 · 2 comments
Open

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

Seibelgys opened this issue Mar 27, 2023 · 2 comments

Comments

@Seibelgys
Copy link

博主您好，我学习了您在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不匹配，想请问博主这种情况怎么办呢
@bubbliiiing
Copy link
Owner

看看

@Seibelgys
Copy link
Author

2
3
1
4

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@bubbliiiing @Seibelgys