Image Segmentation Keras

github.com/divamgupta/image-segmentation-keras

divamgupta/image-segmentation-keras

AttributeError: module 'tensorflow.python.keras.backend' has no attribute 'get_graph'

위 에러가 나온다면 현재 설치되어 있는 keras 버전이 맞지 않은 것 (keras 2.3.1 쓰고 있었는데 에러남.)

 

Anaconda 환경에서 Keras 2.0 설치시 다음을 사용하면 설치가 되지 않음.

$ conda install -c conda-forge keras=2.0.5

 

$ conda search keras

위 명령어로 검색하면 분명 keras 2.0.5가 있기는 있음. 하지만 설치가 되지 않을 것.

이때 다음 명령어로 설치 진행

 

$ pip install Keras==2.0.5

 

만약 기존에 설치되어 있는 keras가 있다면 삭제되고 2.0.5가 설치될 것임.

 

자신이 imageio를 사용하고 있다면 눈물을 머금고 해당 keras 버전으로 찾아서 다시 설치해야할 것ㅠ

 

직접 데이터를 학습하고 싶다면 epoch를 1로 하여 checkpoint를 정상적으로 찾을 수 있는 지 확인부터 할 것.

간혹 학습이 되었음에도 불구하고 checkpoint를 못찾는 문제가 있다.

 

AssertionError: Checkpoint not found.

 

이때는 아래를 참고하자.

 

github.com/divamgupta/image-segmentation-keras/issues/237

Checkpoint is not found · Issue #237 · divamgupta/image-segmentation-keras

Windows에서 실행한다면 다음이 문제일 수도 있다.

 

checkpoints_path="checkpoints\\vgg_unet_1"

 

위와 같이 수정하면 정상적으로 읽어온다.