paperswithcodeをみて、どのdetection modelが良いのかなっと思ってサイトを確認していた。
leaderboardを確認していると、「Simple Copy-Paste is a Strong Data Augmentation Method for Instance Segmentation」というツールがあることがわかり、試してみた。
実験コード
結果
使用方法
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