본문 바로가기
전체보기/머신러닝

Maritime MaskRCNN

by 피망우유 2021. 2. 7.

github.com/Allopart/Maritme_Mask_RCNN

 

Allopart/Maritme_Mask_RCNN

Mask-RCNN applied to DTU Maritime dataset for the autonomous navigation of vessels through image segmentation - Allopart/Maritme_Mask_RCNN

github.com

Anaconda 기준

 

cudatoolkit 9.0

cython 0.29.21

h5py 2.10.0 (3.0 이상 버전 설치 시 .h5 파일을 읽어오지 못하는 현상 발생)

imgaug 0.4.0

keras 2.1.5

pillow 8.0.1

tensorflow-gpu 1.10.0

scikit-image 0.16.2

matplotlib 3.3.3

 

 

labelme 를 이용하여 라벨링된 데이터 필요 (원본 이미지와 .json 파일)

image_list.txt와 labels.txt 파일 생성 필요

python loader.py 를 이용하여 maskrcnn 용으로 데이터 변환 (loader.py 코드 수정 필요)

 

댓글