search

tanakataiki / ssd_kerasV2

BetterNetWork4SSD
MIT License
144 stars 56 forks source link

Got error when training ssd300MobileNet with imagenet #15

Closed hungbui-io closed 5 years ago

hungbui-io commented 5 years ago

Hi @tanakataiki , Thanks a lot for such a great work! I am trying to train your ssd300MobileNet model with ImageNet dataset and got the following error:

ValueError: Error when checking target: expected predictions to have 3 dimensions, but got array with shape (32, 1000)

Do you have idea how to fix this issue? Is there any chance of missing Flatten layers before predictions layer (or after softmax)? Best, Hung

tomsblueskeye commented 5 years ago

How did you solve this issue?