ValueError: Dimension 1 in both shapes must be equal, but are 5 and 3.

      

        使用VGG16进行人群密度等级分类时出现了该问题,原因是之前进行3分类训练得到过一个模型,再进行5分类训练时,(猜测)好像会在之前3分类的模型基础上接着训练,而不是直接训练出一个新的模型进行覆盖。

        解决方法:将文件夹中之前训练出的模型删除掉。

你可能感兴趣的:(深度学习,深度学习,计算机视觉,python,1024程序员节)