paperswithcodeをみて、どのdetection modelが良いのかなっと思ってサイトを確認していた。

f:id:unachan_kz:20210306224732p:plain — Object Detection on COCO test-dev

leaderboardを確認していると、「Simple Copy-Paste is a Strong Data Augmentation Method for Instance Segmentation」というツールがあることがわかり、試してみた。

実験コード

github.com

結果

f:id:unachan_kz:20210306225119p:plain — 魚のaugmentation

f:id:unachan_kz:20210306225246p:plain — 魚のaugmentation

使用方法

pytorchのコードにおいて下記コードを追加すれば良い様だ。

output = transforms(image=image, masks=masks, bboxes=bboxes)
--->instead
output = transforms(
    image=image, masks=masks, bboxes=bboxes,
    paste_image=paste_image, paste_masks=paste_masks, paste_bboxes=paste_bboxes
  )

確認環境

google colaboratory

検証中に以下のエラーが発生しました。

# エラーが出た箇所
from albumentations.pytorch.transforms import ToTensorV2
 
# エラー内容
ImportError: cannot import name 'ToTensorV2' from 'albumentations.pytorch.transforms' (/usr/local/lib/python3.7/dist-packages/albumentations/pytorch/transforms.py)

このため、albumentationsのバージョンを変更しエラーを回避しました。

#! pip install albumentations==0.4.6