├── .gitignore ├── LICENSE ├── NOTICE ├── README.md ├── ann_utils.py ├── checkpoints └── README.md ├── datasets ├── __init__.py ├── pascal2012.py ├── prm_point_dict.json └── proposal_dict.json ├── metrics ├── __init__.py └── ap.py ├── models ├── __init__.py ├── configs │ ├── config2.yaml │ ├── default_config.yaml │ ├── e2e_mask_rcnn_R_101_FPN_1x.yaml │ ├── e2e_mask_rcnn_R_50_FPN_1x.yaml │ ├── e2e_mask_rcnn_R_50_FPN_1x_caffe2.yaml │ ├── fpn_coco_config.yaml │ ├── res_coco_config.yaml │ └── res_config.yaml ├── mrcnn.py └── prm.py ├── results ├── gt_image.png ├── pred_image.png └── pred_image_blacked.png ├── test.py ├── test_on_image.py ├── train.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ServiceNow/wise_ils/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ServiceNow/wise_ils/HEAD/LICENSE -------------------------------------------------------------------------------- /NOTICE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ServiceNow/wise_ils/HEAD/NOTICE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ServiceNow/wise_ils/HEAD/README.md -------------------------------------------------------------------------------- /ann_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ServiceNow/wise_ils/HEAD/ann_utils.py -------------------------------------------------------------------------------- /checkpoints/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ServiceNow/wise_ils/HEAD/checkpoints/README.md -------------------------------------------------------------------------------- /datasets/__init__.py: -------------------------------------------------------------------------------- 1 | from . import pascal2012 2 | 3 | 4 | -------------------------------------------------------------------------------- /datasets/pascal2012.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ServiceNow/wise_ils/HEAD/datasets/pascal2012.py -------------------------------------------------------------------------------- /datasets/prm_point_dict.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ServiceNow/wise_ils/HEAD/datasets/prm_point_dict.json -------------------------------------------------------------------------------- /datasets/proposal_dict.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ServiceNow/wise_ils/HEAD/datasets/proposal_dict.json -------------------------------------------------------------------------------- /metrics/__init__.py: -------------------------------------------------------------------------------- 1 | from . import ap -------------------------------------------------------------------------------- /metrics/ap.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ServiceNow/wise_ils/HEAD/metrics/ap.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | from . import mrcnn -------------------------------------------------------------------------------- /models/configs/config2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ServiceNow/wise_ils/HEAD/models/configs/config2.yaml -------------------------------------------------------------------------------- /models/configs/default_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ServiceNow/wise_ils/HEAD/models/configs/default_config.yaml -------------------------------------------------------------------------------- /models/configs/e2e_mask_rcnn_R_101_FPN_1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ServiceNow/wise_ils/HEAD/models/configs/e2e_mask_rcnn_R_101_FPN_1x.yaml -------------------------------------------------------------------------------- /models/configs/e2e_mask_rcnn_R_50_FPN_1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ServiceNow/wise_ils/HEAD/models/configs/e2e_mask_rcnn_R_50_FPN_1x.yaml -------------------------------------------------------------------------------- /models/configs/e2e_mask_rcnn_R_50_FPN_1x_caffe2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ServiceNow/wise_ils/HEAD/models/configs/e2e_mask_rcnn_R_50_FPN_1x_caffe2.yaml -------------------------------------------------------------------------------- /models/configs/fpn_coco_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ServiceNow/wise_ils/HEAD/models/configs/fpn_coco_config.yaml -------------------------------------------------------------------------------- /models/configs/res_coco_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ServiceNow/wise_ils/HEAD/models/configs/res_coco_config.yaml -------------------------------------------------------------------------------- /models/configs/res_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ServiceNow/wise_ils/HEAD/models/configs/res_config.yaml -------------------------------------------------------------------------------- /models/mrcnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ServiceNow/wise_ils/HEAD/models/mrcnn.py -------------------------------------------------------------------------------- /models/prm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ServiceNow/wise_ils/HEAD/models/prm.py -------------------------------------------------------------------------------- /results/gt_image.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ServiceNow/wise_ils/HEAD/results/gt_image.png -------------------------------------------------------------------------------- /results/pred_image.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ServiceNow/wise_ils/HEAD/results/pred_image.png -------------------------------------------------------------------------------- /results/pred_image_blacked.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ServiceNow/wise_ils/HEAD/results/pred_image_blacked.png -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ServiceNow/wise_ils/HEAD/test.py -------------------------------------------------------------------------------- /test_on_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ServiceNow/wise_ils/HEAD/test_on_image.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ServiceNow/wise_ils/HEAD/train.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ServiceNow/wise_ils/HEAD/utils.py --------------------------------------------------------------------------------