Stinky-Tofu / Stronger-yolo

šŸ”„Improve yolo with latest paper
MIT License
3 stars 0 forks source link

Key conv52/batch_normalization/scale/ExponentialMovingAverage not found in checkpoint #25

Open tienduchoang opened 5 years ago

tienduchoang commented 5 years ago

Hello Stinky-Tofu, I follow your Readme file, and I got this error. I don't know how to fix this. I need your help. Thanks Caused by op 'save/RestoreV2', defined at: File "test.py", line 297, in T = YoloTest() File "test.py", line 49, in init self.__saver = tf.train.Saver(ema_obj.variables_to_restore()) File "/home/tienduchoang/miniconda3/envs/miniconda/lib/python3.5/site-packages/tensorflow/python/training/saver.py", line 832, in init self.build() File "/home/tienduchoang/miniconda3/envs/miniconda/lib/python3.5/site-packages/tensorflow/python/training/saver.py", line 844, in build self._build(self._filename, build_save=True, build_restore=True) File "/home/tienduchoang/miniconda3/envs/miniconda/lib/python3.5/site-packages/tensorflow/python/training/saver.py", line 881, in _build build_save=build_save, build_restore=build_restore) File "/home/tienduchoang/miniconda3/envs/miniconda/lib/python3.5/site-packages/tensorflow/python/training/saver.py", line 513, in _build_internal restore_sequentially, reshape) File "/home/tienduchoang/miniconda3/envs/miniconda/lib/python3.5/site-packages/tensorflow/python/training/saver.py", line 332, in _AddRestoreOps restore_sequentially) File "/home/tienduchoang/miniconda3/envs/miniconda/lib/python3.5/site-packages/tensorflow/python/training/saver.py", line 580, in bulk_restore return io_ops.restore_v2(filename_tensor, names, slices, dtypes) File "/home/tienduchoang/miniconda3/envs/miniconda/lib/python3.5/site-packages/tensorflow/python/ops/gen_io_ops.py", line 1572, in restore_v2 name=name) File "/home/tienduchoang/miniconda3/envs/miniconda/lib/python3.5/site-packages/tensorflow/python/framework/op_def_library.py", line 788, in _apply_op_helper op_def=op_def) File "/home/tienduchoang/miniconda3/envs/miniconda/lib/python3.5/site-packages/tensorflow/python/util/deprecation.py", line 507, in new_func return func(*args, **kwargs) File "/home/tienduchoang/miniconda3/envs/miniconda/lib/python3.5/site-packages/tensorflow/python/framework/ops.py", line 3300, in create_op op_def=op_def) File "/home/tienduchoang/miniconda3/envs/miniconda/lib/python3.5/site-packages/tensorflow/python/framework/ops.py", line 1801, in init self._traceback = tf_stack.extract_stack()

NotFoundError (see above for traceback): Restoring from checkpoint failed. This is most likely due to a Variable name or other graph key that is missing from the checkpoint. Please ensure that you have not altered the graph expected based on the checkpoint. Original error:

Key conv52/batch_normalization/scale/ExponentialMovingAverage not found in checkpoint [[node save/RestoreV2 (defined at test.py:49) ]]

Stinky-Tofu commented 5 years ago

I implemented it with python2, so you can try it by python2