最近在看YOLO算法,在用pycharm跑代码的时候,总是跑不出来,一开始是TensorFlow版本不对,就不断的创建新环境和换tensorflow的各种版本,最后,在python3.6和tensorflow1.2.1的环境下,以前出现的问题都消失了,但是
出现了这个:AttributeError: module ‘tensorflow.python.ops.nn’ has no attribute ‘leaky_relu’。
我在网上看了很多的对于这个问题的解答,都是说Keras的版本不对,要回退成2.1.2版本的,我又修改Keras的版本,发现还是不行。就想放弃跑这个代码了。
在后来的某一天,突然看到了这个https://github.com/keras-team/keras/issues/9349,里面是各国友人对于这个问题的讨论。
这上面说的是将tensorflow的版本更新到1.5.0就没有那个错误了,我就抱着侥幸的心理试了试,发现,成了!我当时欣喜若狂,终于将YOLO代码跑了出来。
好啦,现在来总结一下怎么消除AttributeError: module ‘tensorflow.python.ops.nn’ has no attribute ‘leaky_relu’这个错误,方法就是Keras设为2.1.2版本,tensorflow设为1.5.0版本。
我靠,这么简单居然还花费了我这么多时间,哎,立个帖子希望能帮到像我这样的人吧。