Mask R-CNN 돌려보기

[liv0vil]

TorchVision 객체 검출 미세조정(Finetuning) 튜토리얼
Mask R-CNN Colab Code

[mingxoxo]

베이스라인 github
베이스라인 2/3

[mingxoxo]

파이토치 torchvision의 Mask R-CNN을 활용한 학습용 코드
공식문서 튜토리얼

위의 링크를 통해 mmdetection을 사용하지 않고 자체적으로 구현한 Mask R-CNN 코드 돌려보기

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])

---

코드 돌릴 때 error : colab CUDA error: device-side assert triggered
오류 해결 : len(classes) ---> len(classes)+1 로 변경
class의 종류 갯수에서 + 1을 해주는 이유는 background class를 추가해주기 위함.

model = torchvision.models.detection.maskrcnn_resnet50_fpn(pretrained=True)
in_features = model.roi_heads.box_predictor.cls_score.in_features
model.roi_heads.box_predictor = FastRCNNPredictor(in_features, len(classes)+1)
in_features_mask = model.roi_heads.mask_predictor.conv5_mask.in_channels
hidden_layer = 256
model.roi_heads.mask_predictor = MaskRCNNPredictor(
    in_features_mask, hidden_layer, len(classes)+1)