객체 검출

suhan cho·2022년 4월 11일
0
  • Penn-Fudan Database for Pedestrian Detection and Segmentation 데이터셋으로 미리 학습된 Mask R-CNN모델을 미세조정
  • 인스턴스(이미지 내에서 사람의 위치 좌표와 픽셀 단위의 사람 여부를 구분한 정보 포함) 분할 모델을 학습시킬 것이다.

데이터셋 정의

  • torch.utils.data.Dataset : 데이터셋
  • __len__: 데이터 크기를 반환
  • __getitem__: 데이터 전처리한 후 반환

np.unique
의 모든 고유 값을 검색하고 이러한 고유 값을 정렬합니다.

profile
안녕하세요

0개의 댓글