balancap / SSD-Tensorflow

Single Shot MultiBox Detector in TensorFlow
4.11k stars 1.89k forks source link

ValueError: Cannot reshape a tensor with 3881472 elements to shape [32,84,38,4,21] (8580096 elements) for 'ssd_300_vgg/block4_box/Reshape_1' #384

Open Davdi opened 4 years ago

Davdi commented 4 years ago

hi when in run the commond python train_ssd_network.py --train_dir=${TRAIN_DIR} --dataset_dir=${DATASET_DIR} --dataset_name=pascalvoc_2012 --dataset_split_name=train --model_name=ssd_300_vgg --checkpoint_path=${CHECKPOINT_PATH} --save_summaries_secs=60 --save_interval_secs=600 --weight_decay=0.0005 --optimizer=adam --learning_rate=0.001 --batch_size=32

it shows the error ValueError: Cannot reshape a tensor with 3881472 elements to shape [32,84,38,4,21] (8580096 elements) for 'ssd_300_vgg/block4_box/Reshape_1' (op: 'Reshape') with input shapes: [32,38,38,84], [5] and with input tensors computed as partial shapes: input[1] = [32,84,38,4,21].