search

pierluigiferrari / ssd_keras

A Keras port of Single Shot MultiBox Detector
Apache License 2.0
1.86k stars 935 forks source link

Dimensions unmatch when I was loading pretrained model weight #357

Closed SYH0568 closed 4 years ago

SYH0568 commented 4 years ago

Hello, When I tried to load the VGG_coco_SSD_512.h5 file. I only change the n_classes to 80 of ssd512_inference.ipynb. And I only compile ssd512_inference.ipynb. However I get the error below: Dimensions must be equal, but are 512 and 64 for '{{node conv4_3_norm/mul}} = Mul[T=DT_FLOAT](conv4_3_norm/l2_normalize, conv4_3_norm/mul/ReadVariableOp)' with input shapes: [?,64,64,512], [64]

I have no idea where are the mistakes. Is there something else I should change?

stale[bot] commented 4 years ago

This issue has been automatically marked as stale because it has not had recent activity. It will be closed if no further activity occurs. Thank you for your contributions.