mask rcnn tensorflow >= 2.0
1564
0
혹시 mask rcnn을 tensorflow 버전 2 이상에서 할 수 있는 방법은 없을까요?
직접 코드를 버전2에 맞게 작성하는 게 맞는걸까요???
답변 2
1
Matterport Mask RCNN 패키지를 TF2 환경으로 바꿔서 만든 패키지가 있어서 테스트 해보았습니다.
실습 예제 다 잘 되는데, 아쉽게 Nucleus Segmentation이 학습이 잘 안됩니다. 음.. 데이터가 많은 경우에 문제가 좀 있는것 같습니다. 나머지 inference와 balloon data 학습은 다 잘 됩니다.
소개드릴 github 주소는 아래와 같습니다.
https://github.com/akTwelve/Mask_RCNN
테스트를 colab에 적용한 ipynb 공유 드립니다. 제 github에서 찾으실수 있습니다.
https://github.com/chulminkw/DLCV/tree/master/mask_rcnn_tf2_test
에 tf2를 적용한 실습 코드가 있습니다.