balancap / SSD-Tensorflow

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

I try to train my model with train_ssd_network.py and provide true Datasets-DiR, but return an error #332

Open Plus-Lee opened 5 years ago

Plus-Lee commented 5 years ago

tensorflow.python.framework.errors_impl.FailedPreconditionError: tfrecords; Is a directory [[{{node pascalvoc_2007_data_provider/parallel_read/ReaderReadV2_1}}]] This is my .sh: DATASET_DIR=./tfrecords/ TRAIN_DIR=./log/ python3 train_ssd_network.py \ --train_dir=${TRAIN_DIR} \ --dataset_dir=${DATASET_DIR} \ --dataset_name=pascalvoc_2007 \ --dataset_split_name=train \ --model_name=ssd_300_vgg \ --save_summaries_secs=60 \ --save_interval_secs=600 \ --weight_decay=0.0005 \ --optimizer=adam \ --learning_rate=0.001 \ --batch_size=32

VolleyballBird commented 5 years ago

first run the tf_convert_data.py to produce tfrecords,and then i use the same .sh as your's with the relative path changed to absolute path and it started to train

VolleyballBird commented 5 years ago

but cant get ssd_300_vgg file