반응형
https://github.com/matterport/Mask_RCNN
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 |
댓글