tanakataiki / ssd_kerasV2

BetterNetWork4SSD
MIT License
144 stars 56 forks source link

Error while training with SSD Mobilenet #18

Open bored-wolf opened 5 years ago

bored-wolf commented 5 years ago

When I run the code for training with the SSD Mobilenet model, I get the following error:

ValueError: Shape must be rank 1 but is rank 4 for 'conv1_bn/cond/FusedBatchNorm' (op: 'FusedBatchNorm') with input shapes: [?,150,150,32], [1,150,1,1], [1,150,1,1], [1,150,1,1], [1,150,1,1]

for an input shape of (300,300,3)

Is there any change that would need to be made for rectifying this?