Mask-RCNN出现的问题

出现问题：
ValueError: Dimension 1 in both shapes must be equal, but are 28 and 324. Shapes are [1024,28] and [1024,324]. for ‘Assign_682’ (op: ‘Assign’) with input shapes: [1024,28], [1024,324].

这是由于维度的原因不同所造成。
修改了train.py 的第60行

成功！
如果还未成功，看几个地方是否一样：
1 train.py的 num_classes,
2 model_data 的shape_classes
3. h5文件是否修改。train.py的60行和
4. mask_rcnn中的第18行