pierluigiferrari / ssd_keras

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

error while training with custom dataset in COCO format #395

Closed Avaneesh-S closed 11 months ago

Avaneesh-S commented 11 months ago

Node: 'model/predictions/concat' ConcatOp : Dimension 1 in both shapes must be equal: shape[0] = [32,8732,81] vs. shape[2] = [32,30,8] [[{{node model/predictions/concat}}]] [Op:__inference_train_function_6323]

i am giving only 1 class custom dataset to train the pre trained model from COCO dataset weights , but i run into the above error .