K-Fashion AI 경진대회

알고리즘 | 이미지 | 영상분할 | 이미지 세분화 | mAP

  • moneyIcon 상금 : 1,000만원
  • 753명 마감

 

파이토치를 활용한 Mask R-CNN 학습하기

2020.11.18 14:37 15,330 조회 language

파이토치 torchvision의 Mask R-CNN을 활용한 학습용 코드입니다. 기초적인 부분만 다뤘습니다.  
저희는 mmdetection을 사용하지 않고 자체적으로 구현해서 사용했지만, 
처음 instance segmentation을 해보시는 분, mmdetection을 사용하기 어렵거나  간단한 베이스라인을 구축을 위해
적절하게 수정해서 사용하시면 될 것 같습니다.  

*해당 코드는 torchvision의 input 형태만 확인하고 만든 코드입니다. 
*해당 코드는 1 epoch 정상 작동만 파악하고 업로드한 것을 알려드립니다. 

image: a PIL Image of size (H, W)
target: a dict containing the following fields
    - boxes (FloatTensor[N, 4])
    - labels (Int64Tensor[N])
    - image_id (Int64Tensor[1])
    - area (Tensor[N])
    - iscrowd (UInt8Tensor[N])
    - masks (UInt8Tensor[N, H, W])

코드
로그인이 필요합니다
0 / 1000
Team
2020.11.19 13:32

코드공유 감사합니다. 저는 encoder+FPN 으로 시도해봤는데 mask rcnn에 비해 성능이 한참 좋지못하군요

당쇠
2020.11.19 23:33

파이토치 라이트닝 설문조사하면 SWAG를 보내준다네요

https://bit.ly/35KTRRy

배가_고파졌다
2021.07.17 18:12

감사합니다!

이전 글
이전 글이 존재하지 않습니다.
현재 글
파이토치를 활용한 Mask R-CNN 학습하기
대회 - K-Fashion AI 경진대회
좋아요 24
조회 15,330
댓글 3
3년 전
다음 글
[베이스라인 1/3] 데이터셋의 이해, 데이터 전처리
대회 - K-Fashion AI 경진대회
좋아요 14
조회 7,876
댓글 1
3년 전