Swin Transformer for Object Detection - Inference with COCO dataset 1. Mask RCNN (ResNet50)

on Projects

Swin Transformer for Object Detection - Inference with COCO dataset 1. Mask RCNN (ResNet50)

1. Reference

image
Results and Models Mask R-CNN

2. Checkpoints

mmdetection Mask R-CNN download

Make checkpoints directory at mmdetection.

3. Inference

3.1. Swin Tiny Mask-RCNN

python tools/test.py configs/swin/mask_rcnn_swin-t-p4-w7_fpn_fp16_ms-crop-3x_coco.py checkpoints/mask_rcnn_swin-t-p4-w7_fpn_fp16_ms-crop-3x_coco_20210908_165006-90a4008c.pth --eval bbox segm

Results
image
image

3.2. Swin Small Mask-RCNN

python tools/test.py configs/swin/mask_rcnn_swin-s-p4-w7_fpn_fp16_ms-crop-3x_coco.py checkpoints/mask_rcnn_swin-s-p4-w7_fpn_fp16_ms-crop-3x_coco_20210903_104808-b92c91f1.pth --eval bbox segm

ResultS
image
image