-
Notifications
You must be signed in to change notification settings - Fork 151
A problem occured when training the model #6
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Comments
@amenua did you use the pre-trained model that I provided on the GoogleDrive? please ensure you didn't use the one in balancap's SSD-Tensorflow or the one in tf.slim. The vgg-16 I used here is the reduced-fc version which is faster while maintain comparable accuracy |
@amenua it's seems your pre-trained model created by old version tensorflow ( just only one flie vgg-16.ckpt) ,you should use the model created by new version tensorflow (there have four flies : checkpoint, .ckpt.meta , ckpt.index , ckpt.data......) , or convert the old version model to new version .you can also download the model provided by author. Suggest authors to improve compatibility. @HiKapok |
@poorguy-aos I will generate one checkpoint in v1 version, will update tonight. Thank you! |
This project is not support tensorflow==1.4.0. Some modules of estimator is not integrated. |
@bidai541 please read README more carefully, you should modify some lines in the codes |
Forgive my mistakes @HiKapok |
I have download vgg16, but there still have a problem. I'm using tensorflow 1.8.
raceback (most recent call last):
File "/root/Downloads/SSD.TensorFlow-master/train_ssd.py", line 464, in
tf.app.run()
File "/usr/local/lib/python3.5/dist-packages/tensorflow/python/platform/app.py", line 126, in run
_sys.exit(main(argv))
File "/root/Downloads/SSD.TensorFlow-master/train_ssd.py", line 460, in main
hooks=[logging_hook], max_steps=FLAGS.max_number_of_steps)
File "/usr/local/lib/python3.5/dist-packages/tensorflow/python/estimator/estimator.py", line 363, in train
loss = self._train_model(input_fn, hooks, saving_listeners)
File "/usr/local/lib/python3.5/dist-packages/tensorflow/python/estimator/estimator.py", line 843, in _train_model
return self._train_model_default(input_fn, hooks, saving_listeners)
File "/usr/local/lib/python3.5/dist-packages/tensorflow/python/estimator/estimator.py", line 856, in _train_model_default
features, labels, model_fn_lib.ModeKeys.TRAIN, self.config)
File "/usr/local/lib/python3.5/dist-packages/tensorflow/python/estimator/estimator.py", line 831, in _call_model_fn
model_fn_results = self._model_fn(features=features, **kwargs)
File "/usr/local/lib/python3.5/dist-packages/tensorflow/contrib/estimator/python/estimator/replicate_model_fn.py", line 221, in single_device_model_fn
local_ps_devices=ps_devices)[0] # One device, so one spec is out.
File "/usr/local/lib/python3.5/dist-packages/tensorflow/contrib/estimator/python/estimator/replicate_model_fn.py", line 559, in _get_loss_towers
**optional_params)
File "/root/Downloads/SSD.TensorFlow-master/train_ssd.py", line 403, in ssd_model_fn
scaffold=tf.train.Scaffold(init_fn=get_init_fn()))
File "/root/Downloads/SSD.TensorFlow-master/train_ssd.py", line 158, in get_init_fn
name_remap={'/kernel': '/weights', '/bias': '/biases'})
File "/root/Downloads/SSD.TensorFlow-master/utility/scaffolds.py", line 66, in get_init_fn_for_scaffold
reader = tf.train.NewCheckpointReader(checkpoint_path)
File "/usr/local/lib/python3.5/dist-packages/tensorflow/python/pywrap_tensorflow_internal.py", line 290, in NewCheckpointReader
return CheckpointReader(compat.as_bytes(filepattern), status)
File "/usr/local/lib/python3.5/dist-packages/tensorflow/python/util/compat.py", line 68, in as_bytes
(bytes_or_text,))
TypeError: Expected binary or unicode string, got None
The text was updated successfully, but these errors were encountered: