出现问题:
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行