Skip to content
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

A problem occured when training the model #6

Closed
amenua opened this issue May 22, 2018 · 6 comments
Closed

A problem occured when training the model #6

amenua opened this issue May 22, 2018 · 6 comments

Comments

@amenua
Copy link

amenua commented May 22, 2018

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

@HiKapok
Copy link
Owner

HiKapok commented May 22, 2018

@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

@vitotu
Copy link

vitotu commented May 22, 2018

@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

@HiKapok
Copy link
Owner

HiKapok commented May 22, 2018

@poorguy-aos I will generate one checkpoint in v1 version, will update tonight. Thank you!

@bidai541
Copy link

This project is not support tensorflow==1.4.0. Some modules of estimator is not integrated.
tf.contrib.estimator.TowerOptimizer
tf.contrib.estimator.replicate_model_fn
It's better to specify the version of tensorflow.

@HiKapok
Copy link
Owner

HiKapok commented May 22, 2018

@bidai541 please read README more carefully, you should modify some lines in the codes

@bidai541
Copy link

Forgive my mistakes @HiKapok

@HiKapok HiKapok closed this as completed May 22, 2018
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

4 participants