在学习Mask_Rcnn框架中,在进行train命令训练过程中,出现了错误ValueError: Dimension 1 in both shapes must be equal, but are 8 and 324 for 'Assign_682' (op: 'Assign') with input shapes: [1024,8], [1024,324].
通过查找与查询,在一个大佬那里解决了疑惑:即官方的h5总共是80类,所以需要在代码中将
NUM_CLASSES = 1 + 1;
改为
NUM_CLASSES = 1 + 80.
最后解决问题。