とりあえず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
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