Stinky-Tofu / Stronger-yolo

🔥Improve yolo with latest paper
MIT License
3 stars 0 forks source link

error exits when rename.py is run #108

Open jwnirvana opened 4 years ago

jwnirvana commented 4 years ago

很棒的工作,但是我遇到个小问题,搞了很久了

(v1) zsl@user:~/wj/v1$ python rename.py Traceback (most recent call last): File "rename.py", line 14, in load = tf.train.import_meta_graph(org_weights_path + '.meta') File "/home/zhaoshiliang/miniconda3/envs/v1/lib/python2.7/site-packages/tensorflow/python/training/saver.py", line 1955, in import_meta_graph *kwargs) File "/home/zhaoshiliang/miniconda3/envs/v1/lib/python2.7/site-packages/tensorflow/python/framework/meta_graph.py", line 743, in import_scoped_meta_graph producer_op_list=producer_op_list) File "/home/zhaoshiliang/miniconda3/envs/v1/lib/python2.7/site-packages/tensorflow/python/util/deprecation.py", line 432, in new_func return func(args, **kwargs) File "/home/zhaoshiliang/miniconda3/envs/v1/lib/python2.7/site-packages/tensorflow/python/framework/importer.py", line 460, in import_graph_def _RemoveDefaultAttrs(op_dict, producer_op_list, graph_def) File "/home/zhaoshiliang/miniconda3/envs/v1/lib/python2.7/site-packages/tensorflow/python/framework/importer.py", line 227, in _RemoveDefaultAttrs op_def = op_dict[node.op] KeyError: u'LeakyRelu'

我把环境搭的合作者一模一样了。我搞了好久了,希望有人能告知我怎么改,或者能分享一下rename后的预训练模型也可以,谢谢!