detectron2 の履歴(No.8)

detectron2の物体検出系で学習させるデータは，基本的に，Cocoフォーマットか？
本研究室の場合，labelmeで作成したjsonファイルからCocoフォーマットへの変換がよくある流れ．
labelme/examples/instance_segmentation/labelme2coco.py

Mask R-CNN†

demoフォルダにあるdemo.pyを使用する．
そしてModel Zooがいい。
設定ファイル（yaml形式）たちはconfigsにある。

python demo.py --config-file ../configs/COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x.yaml --video-input /mnt/c/Users/survey/Documents/GitHub/detectron2_wkk/demo/GX070226s.mp4 --output /mnt/c/Users/survey/Documents/GitHub/detectron2_wkk/demo/outputs2.mp4 --opts MODEL.WEIGHTS detectron2://COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x/137849600/model_final_f10217.pkl

Cityscapes (Mask R-CNN)

python demo.py --config-file ../configs/Cityscapes/mask_rcnn_R_50_FPN.yaml --video-input /mnt/c/Users/survey/Documents/GitHub/detectron2_wkk/video_nakatsugawa.mp4 --output /mnt/c/Users/survey/Documents/GitHub/detectron2_wkk/demo/nakatsu_cityscape_outputs.mp4 --opts MODEL.WEIGHTS detectron2://COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x/137849600/model_final_f10217.pkl

python tools/train_net.py --eval-only --config-file configs/Cityscapes/mask_rcnn_R_50_FPN.yaml MODEL.WEIGHTS https://dl.fbaipublicfiles.com/detectron2/Cityscapes/mask_rcnn_R_50_FPN/142423278/model_final_af9cf5.pkl

↑

Faster R-CNN†

動かしたいが，情報が少ない．．

python demo.py --config-file ../configs/COCO-Detection/faster_rcnn_R_50_FPN_3x.yaml --video-input /mnt/c/Users/survey/Documents/GitHub/detectron2_wkk/demo/video_nakatsugawa.mp4 --output /mnt/c/Users/survey/Documents/GitHub/detectron2_wkk/demo/video_nakatsugawa_output.mp4 --opts MODEL.WEIGHTS detectron2://COCO-Detection/faster_rcnn_R_50_FPN_3x/137849458/model_final_280758.pkl

↑

転移学習†

転移学習:tools/train_net.pyを使えばいいみたい．詳細は転移学習-cityscapes, cityscapesscripts, Mask R-CNN, Detectron2

AI Tips