본문 바로가기
인공지능/CV

MASK RCNN 실행시 버전오류

by EXUPERY 2021. 4. 20.
반응형

https://github.com/matterport/Mask_RCNN

 

matterport/Mask_RCNN

Mask R-CNN for object detection and instance segmentation on Keras and TensorFlow - matterport/Mask_RCNN

github.com

MASK RCNN에서 실습을 위해 위 레포를 클론하였습니다. 환경은 Colab입니다.

코랩에서 git clone을 했을때의 화면입니다.

루트 디렉토리를 코랩환경에 맞게 다시 설정해주어야 합니다. 루트디렉토리가 맞지 않으면 모듈을 불러올 수 없습니다.

모델을 객체화하는데 문제가 생겼습니다. tensorflow에는 log라는 속성이 없습니다. 1년전에 작성된 레포이기때문에 버전문제라고 짐작할 수 있습니다. 

requirements.txt입니다. tensorflow와 keras의 버전이 위와 같습니다. 1년 동안 텐서플로우와 케라스에서 업데이트가 굉장히 많았습니다. 코랩환경인것을 감안하여 다음과 같이 버전을 낮춰주었습니다.

pip install tensorflow==1.13.1
pip install keras==2.1.5

다시 실행한 결과는 다음과 같습니다.

경고문이 뜨긴하지만 정상적으로 작동하는 것을 볼 수 있습니다. 경고문에는 이 후 버전에서는 제거되는 모듈이라고 설명되어있습니다. 버전만바꾸어도 잘 작동해서 다행입니다.

 

이후 모델을 실행시키면 결과를 확인할 수 있습니다.

 

반응형

'인공지능 > CV' 카테고리의 다른 글

객체탐지 (Object Detection) 2. YOLO !! (v1~v3)  (2) 2021.05.01
객체탐지 (Object Detection) 1. YOLO 이전 까지 흐름  (0) 2021.05.01
[CODE] EfficientNetB7  (0) 2021.04.10
[CODE] VGG19  (0) 2021.04.10
[CODE] ANN, K-fold CV, CNN  (0) 2021.04.10

댓글