ValueError: Dimension 1 in both shapes must be equal, but are 324 and 8

[Abduoit]

Hi

Thanks for your effort, I used visualize_cv.py it works properly. But when I change the file mask_rcnn_coco.h5 with my own .h5 that I trained, it gives me the following error.

any help please ??

ValueError: Dimension 1 in both shapes must be equal, but are 324 and 8. Shapes are [1024,324] and [1024,8]. for 'Assign_682' (op: 'Assign') with input shapes: [1024,324], [1024,8].

[markjay4k]

I'm not sure why that is. I haven't tried training a mask RCNN model before

[anburiyan]

I have the similar issue when i using pre-trained coco model. I know we should change the image input size but am not sure in config file where i have to change to resolve the issue

[Abduoit]

@anburiyan did you solve your issue ?

[anburiyan]

@Abduoit - not yet. But, I used existing model which worked well. For my own trained model still getting.