pspnet训练时报错Check failed: outer_num_ * inner_num_ == bottom[1]->count()

爱学习的人工智障

于 2019-05-21 15:00:14 发布

阅读量641

点赞数

CC 4.0 BY-SA版权

分类专栏：深度学习 caffe

本文链接：https://blog.youkuaiyun.com/weixin_38410551/article/details/90407048

深度学习同时被 2 个专栏收录

107 篇文章 ¥9.90 ¥99.00

订阅专栏

超级会员免费看

caffe

21 篇文章

订阅专栏

在使用PSPNet进行语义分割训练时遇到'Check failed: outer_num_ * inner_num_ == bottom[1]->count()'的错误，该问题通常与输入数据维度不匹配有关。检查数据预处理步骤，确保输入图像和标签的尺寸一致，并正确调整网络配置文件中的参数。此外，检查批大小设置是否正确，避免批次过大导致内存溢出。通过对代码审查和参数调整，可以解决此错误并继续训练过程。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Check failed: outer_num_ * inner_num_ == bottom[1]->count() (497025 vs. 504100) Number of labels must match number of predictions; e.g., if softmax axis == 1 and prediction shape is (N, C, H, W), label count (number of labels) must be N*H*W, with integer values in {0, 1, ..., C-1}.
问题大概是说，prediction 与label的shape不一致，bottom[1]->count()就是说的label的count等于504100，prediction的outer_num_ * inner_num_等于497025，也就是说label与prediction不一致。
解决办法：需要去理解每一层的网络的输出，分析是哪一层的输出导致出现的问题。
参考链接：
https://blog.youkuaiyun.com/qq_30024069/article/details/85312170