예외가 발생했습니다. AttributeError
module 'tensorflow' has no attribute 'GraphDef'
File "G:\office\enterprise_sdk_int_g_ui\develnote\deeplearning\cnn-colorize-master (1)\cnn-colorize-master\colorize.py", line 54, in main
graph_def = tf.GraphDef()
File "G:\office\enterprise_sdk_int_g_ui\develnote\deeplearning\cnn-colorize-master (1)\cnn-colorize-master\colorize.py", line 75, in <module>
main()
원인은 tensorflow-gpu 가 설치되지 않아서 발생한다. pip 명령으로 tensorflow-gpu 를 설치한다.
또는 tensorflow 버전에 따라서 다를 수 있습니다.
tf.compat.v1.GraphDef() # -> instead of tf.GraphDef()
tf.compat.v2.io.gfile.GFile() # -> instead of tf.gfile.GFile()
Colorization (0) | 2021.04.02 |
---|---|
VGG-16-Gray (0) | 2021.03.30 |
Logistic Regression - 회귀 (0) | 2016.01.26 |
의사결정트리(Decision Tree) (0) | 2016.01.26 |
Nearest Neighbor Classification (0) | 2016.01.19 |
댓글 영역