- Penn-Fudan Database for Pedestrian Detection and Segmentation 데이터셋으로 미리 학습된 Mask R-CNN모델을 미세조정
- 인스턴스(이미지 내에서 사람의 위치 좌표와 픽셀 단위의 사람 여부를 구분한 정보 포함) 분할 모델을 학습시킬 것이다.
데이터셋 정의
- torch.utils.data.Dataset : 데이터셋
- __len__: 데이터 크기를 반환
- __getitem__: 데이터 전처리한 후 반환
np.unique
의 모든 고유 값을 검색하고 이러한 고유 값을 정렬합니다.