nyxrobotics / annotateFromVideo

It is an annotation tool for create training data of Instance segmentation, such as Mask R-CNN or YOLACT

annotateFromVideo

これはMask R-CNNやYOLACTなどのInstance segmentationの、訓練データ生成用のアノテーションツールです。 It is an annotation tool for create training data of Instance segmentation, such as Mask R-CNN or YOLACT

https://youtu.be/Z1RKNU1CqGM

私の環境は下記の通りですが、違っていても動くと思います。

• Python : 3.8
• numpy : 1.16
• opencv-python : 4.1.1

実行するためのコマンドは下記の通りです。

python annotateFromVideo.py targetVideo.mp4 targetObjectName

targetVideo.mp4には、Annotationしたい動画のファイル名を、
targetObjectNameには、対象物体の名称を記述してください。

実行中は下記のことができます。

1. [ b ]必ず背景としたい箇所をマークできるモードに変更
2. [ w ]必ず前景としたい箇所をマークできるモードに変更
3. [ g ]背景と前景の抽出を試行する
4. [esc]背景と前景の抽出を確定し、次の画像へ進む
5. [ q ]途中でツールを終了できる

～注意点～

• アノテーションをやり直したい場合、その画像のjsonファイルを削除し、プログラムを実行しなおすと再挑戦できます。
• outputResultsなどのフォルダがない場合、上手く実行できない可能性があります。
• データ拡張の枚数を増やしたい場合は、394行目を変更してください。