├── .gitignore ├── LICENSE ├── README.md ├── datasets ├── __init__.py ├── coco.py ├── data_aug.py ├── lvis.py ├── objects365.py ├── pl_data_module.py ├── voc.py └── ytvis.py ├── dockerfile ├── main.py ├── mmdet ├── configs │ ├── GTMask │ │ ├── cascade_mask_rcnn_convnext-b_p4_w7_fpn_giou_4conv1f_fp16_ms-crop_3x_coco.py │ │ ├── cascade_mask_rcnn_convnext-s_p4_w7_fpn_giou_4conv1f_fp16_ms-crop_3x_coco.py │ │ ├── cascade_mask_rcnn_convnext-t_p4_w7_fpn_giou_4conv1f_fp16_ms-crop_3x_coco.py │ │ ├── mask2former_r101_lsj_8x2_50e_coco-panoptic.py │ │ ├── mask2former_r101_lsj_8x2_50e_coco.py │ │ ├── mask2former_r50_lsj_8x2_50e_coco-panoptic.py │ │ ├── mask2former_r50_lsj_8x2_50e_coco.py │ │ ├── mask2former_swin-b-p4-w12-384-in21k_lsj_8x2_50e_coco-panoptic.py │ │ ├── mask2former_swin-b-p4-w12-384_lsj_8x2_50e_coco-panoptic.py │ │ ├── mask2former_swin-l-p4-w12-384-in21k_lsj_16x1_100e_coco-panoptic.py │ │ ├── mask2former_swin-s-p4-w7-224_lsj_8x2_50e_coco-panoptic.py │ │ ├── mask2former_swin-s-p4-w7-224_lsj_8x2_50e_coco.py │ │ ├── mask2former_swin-t-p4-w7-224_lsj_8x2_50e_coco-panoptic.py │ │ ├── mask2former_swin-t-p4-w7-224_lsj_8x2_50e_coco.py │ │ ├── mask_rcnn_convnext-t_p4_w7_fpn_fp16_ms-crop_3x_coco.py │ │ ├── solov2_light_r18_fpn_3x_coco.py │ │ ├── solov2_light_r34_fpn_3x_coco.py │ │ ├── solov2_light_r50_dcn_fpn_3x_coco.py │ │ ├── solov2_light_r50_fpn_3x_coco.py │ │ ├── solov2_r101_dcn_fpn_3x_coco.py │ │ ├── solov2_r101_fpn_3x_coco.py │ │ ├── solov2_r50_fpn_1x_coco.py │ │ ├── solov2_r50_fpn_3x_coco.py │ │ └── solov2_x101_dcn_fpn_3x_coco.py │ ├── MALMask │ │ ├── cascade_mask_rcnn_convnext-b_p4_w7_fpn_giou_4conv1f_fp16_ms-crop_3x_coco_mal.py │ │ ├── cascade_mask_rcnn_convnext-s_p4_w7_fpn_giou_4conv1f_fp16_ms-crop_3x_coco_mal.py │ │ ├── cascade_mask_rcnn_convnext-s_p4_w7_fpn_giou_4conv1f_fp16_ms-crop_3x_coco_mal_obj365.py │ │ ├── mask2former_swin-s-p4-w7-224_lsj_8x2_50e_coco_mal.py │ │ ├── mask_rcnn_r101_fpn_sample1e-3_mstrain_1x_lvis_v1_mal.py │ │ ├── mask_rcnn_x101_32x4d_fpn_sample1e-3_mstrain_1x_lvis_v1_mal.py │ │ ├── solov2_r101_dcn_fpn_3x_coco_mal.py │ │ ├── solov2_r50_fpn_3x_coco_mal.py │ │ └── solov2_x101_dcn_fpn_3x_coco_mal.py │ └── _base_ │ │ ├── datasets │ │ ├── cityscapes_detection.py │ │ ├── cityscapes_instance.py │ │ ├── coco_detection.py │ │ ├── coco_instance.py │ │ ├── coco_instance_essenco_vit_mae_base.py │ │ ├── coco_instance_semantic.py │ │ ├── coco_panoptic.py │ │ ├── deepfashion.py │ │ ├── lvis_v0.5_instance.py │ │ ├── lvis_v1_instance.py │ │ ├── openimages_detection.py │ │ ├── voc0712.py │ │ └── wider_face.py │ │ ├── default_runtime.py │ │ ├── models │ │ ├── cascade_mask_rcnn_r50_fpn.py │ │ ├── cascade_rcnn_r50_fpn.py │ │ ├── fast_rcnn_r50_fpn.py │ │ ├── faster_rcnn_r50_caffe_c4.py │ │ ├── faster_rcnn_r50_caffe_dc5.py │ │ ├── faster_rcnn_r50_fpn.py │ │ ├── mask_rcnn_r50_caffe_c4.py │ │ ├── mask_rcnn_r50_fpn.py │ │ ├── retinanet_r50_fpn.py │ │ ├── rpn_r50_caffe_c4.py │ │ ├── rpn_r50_fpn.py │ │ └── ssd300.py │ │ └── schedules │ │ ├── schedule_1x.py │ │ ├── schedule_20e.py │ │ └── schedule_2x.py └── tools │ ├── analysis_tools │ ├── analyze_logs.py │ ├── analyze_results.py │ ├── benchmark.py │ ├── coco_error_analysis.py │ ├── confusion_matrix.py │ ├── eval_metric.py │ ├── get_flops.py │ ├── optimize_anchors.py │ ├── robustness_eval.py │ └── test_robustness.py │ ├── dataset_converters │ ├── cityscapes.py │ ├── images2coco.py │ └── pascal_voc.py │ ├── deployment │ ├── mmdet2torchserve.py │ ├── mmdet_handler.py │ ├── onnx2tensorrt.py │ ├── pytorch2onnx.py │ ├── test.py │ └── test_torchserver.py │ ├── dist_test.sh │ ├── dist_train.sh │ ├── misc │ ├── browse_dataset.py │ ├── download_dataset.py │ ├── gen_coco_panoptic_test_info.py │ ├── get_image_metas.py │ ├── print_config.py │ └── split_coco.py │ ├── model_converters │ ├── detectron2pytorch.py │ ├── publish_model.py │ ├── regnet2mmdet.py │ ├── selfsup2mmdet.py │ ├── upgrade_model_version.py │ └── upgrade_ssd_version.py │ ├── slurm_test.sh │ ├── slurm_train.sh │ ├── test.py │ └── train.py ├── models ├── __init__.py ├── mal.py └── vision_transformers.py ├── tools ├── convert2lmdb.py ├── move_objects365_data.sh ├── mv_data_from_ddn.py ├── visualize_anns.py └── visualize_attention.py └── utils ├── __init__.py └── optimizers └── adamw.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/README.md -------------------------------------------------------------------------------- /datasets/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /datasets/coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/datasets/coco.py -------------------------------------------------------------------------------- /datasets/data_aug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/datasets/data_aug.py -------------------------------------------------------------------------------- /datasets/lvis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/datasets/lvis.py -------------------------------------------------------------------------------- /datasets/objects365.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/datasets/objects365.py -------------------------------------------------------------------------------- /datasets/pl_data_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/datasets/pl_data_module.py -------------------------------------------------------------------------------- /datasets/voc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/datasets/voc.py -------------------------------------------------------------------------------- /datasets/ytvis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/datasets/ytvis.py -------------------------------------------------------------------------------- /dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/dockerfile -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/main.py -------------------------------------------------------------------------------- /mmdet/configs/GTMask/cascade_mask_rcnn_convnext-b_p4_w7_fpn_giou_4conv1f_fp16_ms-crop_3x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/configs/GTMask/cascade_mask_rcnn_convnext-b_p4_w7_fpn_giou_4conv1f_fp16_ms-crop_3x_coco.py -------------------------------------------------------------------------------- /mmdet/configs/GTMask/cascade_mask_rcnn_convnext-s_p4_w7_fpn_giou_4conv1f_fp16_ms-crop_3x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/configs/GTMask/cascade_mask_rcnn_convnext-s_p4_w7_fpn_giou_4conv1f_fp16_ms-crop_3x_coco.py -------------------------------------------------------------------------------- /mmdet/configs/GTMask/cascade_mask_rcnn_convnext-t_p4_w7_fpn_giou_4conv1f_fp16_ms-crop_3x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/configs/GTMask/cascade_mask_rcnn_convnext-t_p4_w7_fpn_giou_4conv1f_fp16_ms-crop_3x_coco.py -------------------------------------------------------------------------------- /mmdet/configs/GTMask/mask2former_r101_lsj_8x2_50e_coco-panoptic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/configs/GTMask/mask2former_r101_lsj_8x2_50e_coco-panoptic.py -------------------------------------------------------------------------------- /mmdet/configs/GTMask/mask2former_r101_lsj_8x2_50e_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/configs/GTMask/mask2former_r101_lsj_8x2_50e_coco.py -------------------------------------------------------------------------------- /mmdet/configs/GTMask/mask2former_r50_lsj_8x2_50e_coco-panoptic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/configs/GTMask/mask2former_r50_lsj_8x2_50e_coco-panoptic.py -------------------------------------------------------------------------------- /mmdet/configs/GTMask/mask2former_r50_lsj_8x2_50e_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/configs/GTMask/mask2former_r50_lsj_8x2_50e_coco.py -------------------------------------------------------------------------------- /mmdet/configs/GTMask/mask2former_swin-b-p4-w12-384-in21k_lsj_8x2_50e_coco-panoptic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/configs/GTMask/mask2former_swin-b-p4-w12-384-in21k_lsj_8x2_50e_coco-panoptic.py -------------------------------------------------------------------------------- /mmdet/configs/GTMask/mask2former_swin-b-p4-w12-384_lsj_8x2_50e_coco-panoptic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/configs/GTMask/mask2former_swin-b-p4-w12-384_lsj_8x2_50e_coco-panoptic.py -------------------------------------------------------------------------------- /mmdet/configs/GTMask/mask2former_swin-l-p4-w12-384-in21k_lsj_16x1_100e_coco-panoptic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/configs/GTMask/mask2former_swin-l-p4-w12-384-in21k_lsj_16x1_100e_coco-panoptic.py -------------------------------------------------------------------------------- /mmdet/configs/GTMask/mask2former_swin-s-p4-w7-224_lsj_8x2_50e_coco-panoptic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/configs/GTMask/mask2former_swin-s-p4-w7-224_lsj_8x2_50e_coco-panoptic.py -------------------------------------------------------------------------------- /mmdet/configs/GTMask/mask2former_swin-s-p4-w7-224_lsj_8x2_50e_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/configs/GTMask/mask2former_swin-s-p4-w7-224_lsj_8x2_50e_coco.py -------------------------------------------------------------------------------- /mmdet/configs/GTMask/mask2former_swin-t-p4-w7-224_lsj_8x2_50e_coco-panoptic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/configs/GTMask/mask2former_swin-t-p4-w7-224_lsj_8x2_50e_coco-panoptic.py -------------------------------------------------------------------------------- /mmdet/configs/GTMask/mask2former_swin-t-p4-w7-224_lsj_8x2_50e_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/configs/GTMask/mask2former_swin-t-p4-w7-224_lsj_8x2_50e_coco.py -------------------------------------------------------------------------------- /mmdet/configs/GTMask/mask_rcnn_convnext-t_p4_w7_fpn_fp16_ms-crop_3x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/configs/GTMask/mask_rcnn_convnext-t_p4_w7_fpn_fp16_ms-crop_3x_coco.py -------------------------------------------------------------------------------- /mmdet/configs/GTMask/solov2_light_r18_fpn_3x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/configs/GTMask/solov2_light_r18_fpn_3x_coco.py -------------------------------------------------------------------------------- /mmdet/configs/GTMask/solov2_light_r34_fpn_3x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/configs/GTMask/solov2_light_r34_fpn_3x_coco.py -------------------------------------------------------------------------------- /mmdet/configs/GTMask/solov2_light_r50_dcn_fpn_3x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/configs/GTMask/solov2_light_r50_dcn_fpn_3x_coco.py -------------------------------------------------------------------------------- /mmdet/configs/GTMask/solov2_light_r50_fpn_3x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/configs/GTMask/solov2_light_r50_fpn_3x_coco.py -------------------------------------------------------------------------------- /mmdet/configs/GTMask/solov2_r101_dcn_fpn_3x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/configs/GTMask/solov2_r101_dcn_fpn_3x_coco.py -------------------------------------------------------------------------------- /mmdet/configs/GTMask/solov2_r101_fpn_3x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/configs/GTMask/solov2_r101_fpn_3x_coco.py -------------------------------------------------------------------------------- /mmdet/configs/GTMask/solov2_r50_fpn_1x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/configs/GTMask/solov2_r50_fpn_1x_coco.py -------------------------------------------------------------------------------- /mmdet/configs/GTMask/solov2_r50_fpn_3x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/configs/GTMask/solov2_r50_fpn_3x_coco.py -------------------------------------------------------------------------------- /mmdet/configs/GTMask/solov2_x101_dcn_fpn_3x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/configs/GTMask/solov2_x101_dcn_fpn_3x_coco.py -------------------------------------------------------------------------------- /mmdet/configs/MALMask/cascade_mask_rcnn_convnext-b_p4_w7_fpn_giou_4conv1f_fp16_ms-crop_3x_coco_mal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/configs/MALMask/cascade_mask_rcnn_convnext-b_p4_w7_fpn_giou_4conv1f_fp16_ms-crop_3x_coco_mal.py -------------------------------------------------------------------------------- /mmdet/configs/MALMask/cascade_mask_rcnn_convnext-s_p4_w7_fpn_giou_4conv1f_fp16_ms-crop_3x_coco_mal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/configs/MALMask/cascade_mask_rcnn_convnext-s_p4_w7_fpn_giou_4conv1f_fp16_ms-crop_3x_coco_mal.py -------------------------------------------------------------------------------- /mmdet/configs/MALMask/cascade_mask_rcnn_convnext-s_p4_w7_fpn_giou_4conv1f_fp16_ms-crop_3x_coco_mal_obj365.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/configs/MALMask/cascade_mask_rcnn_convnext-s_p4_w7_fpn_giou_4conv1f_fp16_ms-crop_3x_coco_mal_obj365.py -------------------------------------------------------------------------------- /mmdet/configs/MALMask/mask2former_swin-s-p4-w7-224_lsj_8x2_50e_coco_mal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/configs/MALMask/mask2former_swin-s-p4-w7-224_lsj_8x2_50e_coco_mal.py -------------------------------------------------------------------------------- /mmdet/configs/MALMask/mask_rcnn_r101_fpn_sample1e-3_mstrain_1x_lvis_v1_mal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/configs/MALMask/mask_rcnn_r101_fpn_sample1e-3_mstrain_1x_lvis_v1_mal.py -------------------------------------------------------------------------------- /mmdet/configs/MALMask/mask_rcnn_x101_32x4d_fpn_sample1e-3_mstrain_1x_lvis_v1_mal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/configs/MALMask/mask_rcnn_x101_32x4d_fpn_sample1e-3_mstrain_1x_lvis_v1_mal.py -------------------------------------------------------------------------------- /mmdet/configs/MALMask/solov2_r101_dcn_fpn_3x_coco_mal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/configs/MALMask/solov2_r101_dcn_fpn_3x_coco_mal.py -------------------------------------------------------------------------------- /mmdet/configs/MALMask/solov2_r50_fpn_3x_coco_mal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/configs/MALMask/solov2_r50_fpn_3x_coco_mal.py -------------------------------------------------------------------------------- /mmdet/configs/MALMask/solov2_x101_dcn_fpn_3x_coco_mal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/configs/MALMask/solov2_x101_dcn_fpn_3x_coco_mal.py -------------------------------------------------------------------------------- /mmdet/configs/_base_/datasets/cityscapes_detection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/configs/_base_/datasets/cityscapes_detection.py -------------------------------------------------------------------------------- /mmdet/configs/_base_/datasets/cityscapes_instance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/configs/_base_/datasets/cityscapes_instance.py -------------------------------------------------------------------------------- /mmdet/configs/_base_/datasets/coco_detection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/configs/_base_/datasets/coco_detection.py -------------------------------------------------------------------------------- /mmdet/configs/_base_/datasets/coco_instance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/configs/_base_/datasets/coco_instance.py -------------------------------------------------------------------------------- /mmdet/configs/_base_/datasets/coco_instance_essenco_vit_mae_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/configs/_base_/datasets/coco_instance_essenco_vit_mae_base.py -------------------------------------------------------------------------------- /mmdet/configs/_base_/datasets/coco_instance_semantic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/configs/_base_/datasets/coco_instance_semantic.py -------------------------------------------------------------------------------- /mmdet/configs/_base_/datasets/coco_panoptic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/configs/_base_/datasets/coco_panoptic.py -------------------------------------------------------------------------------- /mmdet/configs/_base_/datasets/deepfashion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/configs/_base_/datasets/deepfashion.py -------------------------------------------------------------------------------- /mmdet/configs/_base_/datasets/lvis_v0.5_instance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/configs/_base_/datasets/lvis_v0.5_instance.py -------------------------------------------------------------------------------- /mmdet/configs/_base_/datasets/lvis_v1_instance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/configs/_base_/datasets/lvis_v1_instance.py -------------------------------------------------------------------------------- /mmdet/configs/_base_/datasets/openimages_detection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/configs/_base_/datasets/openimages_detection.py -------------------------------------------------------------------------------- /mmdet/configs/_base_/datasets/voc0712.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/configs/_base_/datasets/voc0712.py -------------------------------------------------------------------------------- /mmdet/configs/_base_/datasets/wider_face.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/configs/_base_/datasets/wider_face.py -------------------------------------------------------------------------------- /mmdet/configs/_base_/default_runtime.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/configs/_base_/default_runtime.py -------------------------------------------------------------------------------- /mmdet/configs/_base_/models/cascade_mask_rcnn_r50_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/configs/_base_/models/cascade_mask_rcnn_r50_fpn.py -------------------------------------------------------------------------------- /mmdet/configs/_base_/models/cascade_rcnn_r50_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/configs/_base_/models/cascade_rcnn_r50_fpn.py -------------------------------------------------------------------------------- /mmdet/configs/_base_/models/fast_rcnn_r50_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/configs/_base_/models/fast_rcnn_r50_fpn.py -------------------------------------------------------------------------------- /mmdet/configs/_base_/models/faster_rcnn_r50_caffe_c4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/configs/_base_/models/faster_rcnn_r50_caffe_c4.py -------------------------------------------------------------------------------- /mmdet/configs/_base_/models/faster_rcnn_r50_caffe_dc5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/configs/_base_/models/faster_rcnn_r50_caffe_dc5.py -------------------------------------------------------------------------------- /mmdet/configs/_base_/models/faster_rcnn_r50_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/configs/_base_/models/faster_rcnn_r50_fpn.py -------------------------------------------------------------------------------- /mmdet/configs/_base_/models/mask_rcnn_r50_caffe_c4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/configs/_base_/models/mask_rcnn_r50_caffe_c4.py -------------------------------------------------------------------------------- /mmdet/configs/_base_/models/mask_rcnn_r50_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/configs/_base_/models/mask_rcnn_r50_fpn.py -------------------------------------------------------------------------------- /mmdet/configs/_base_/models/retinanet_r50_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/configs/_base_/models/retinanet_r50_fpn.py -------------------------------------------------------------------------------- /mmdet/configs/_base_/models/rpn_r50_caffe_c4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/configs/_base_/models/rpn_r50_caffe_c4.py -------------------------------------------------------------------------------- /mmdet/configs/_base_/models/rpn_r50_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/configs/_base_/models/rpn_r50_fpn.py -------------------------------------------------------------------------------- /mmdet/configs/_base_/models/ssd300.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/configs/_base_/models/ssd300.py -------------------------------------------------------------------------------- /mmdet/configs/_base_/schedules/schedule_1x.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/configs/_base_/schedules/schedule_1x.py -------------------------------------------------------------------------------- /mmdet/configs/_base_/schedules/schedule_20e.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/configs/_base_/schedules/schedule_20e.py -------------------------------------------------------------------------------- /mmdet/configs/_base_/schedules/schedule_2x.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/configs/_base_/schedules/schedule_2x.py -------------------------------------------------------------------------------- /mmdet/tools/analysis_tools/analyze_logs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/tools/analysis_tools/analyze_logs.py -------------------------------------------------------------------------------- /mmdet/tools/analysis_tools/analyze_results.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/tools/analysis_tools/analyze_results.py -------------------------------------------------------------------------------- /mmdet/tools/analysis_tools/benchmark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/tools/analysis_tools/benchmark.py -------------------------------------------------------------------------------- /mmdet/tools/analysis_tools/coco_error_analysis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/tools/analysis_tools/coco_error_analysis.py -------------------------------------------------------------------------------- /mmdet/tools/analysis_tools/confusion_matrix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/tools/analysis_tools/confusion_matrix.py -------------------------------------------------------------------------------- /mmdet/tools/analysis_tools/eval_metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/tools/analysis_tools/eval_metric.py -------------------------------------------------------------------------------- /mmdet/tools/analysis_tools/get_flops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/tools/analysis_tools/get_flops.py -------------------------------------------------------------------------------- /mmdet/tools/analysis_tools/optimize_anchors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/tools/analysis_tools/optimize_anchors.py -------------------------------------------------------------------------------- /mmdet/tools/analysis_tools/robustness_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/tools/analysis_tools/robustness_eval.py -------------------------------------------------------------------------------- /mmdet/tools/analysis_tools/test_robustness.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/tools/analysis_tools/test_robustness.py -------------------------------------------------------------------------------- /mmdet/tools/dataset_converters/cityscapes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/tools/dataset_converters/cityscapes.py -------------------------------------------------------------------------------- /mmdet/tools/dataset_converters/images2coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/tools/dataset_converters/images2coco.py -------------------------------------------------------------------------------- /mmdet/tools/dataset_converters/pascal_voc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/tools/dataset_converters/pascal_voc.py -------------------------------------------------------------------------------- /mmdet/tools/deployment/mmdet2torchserve.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/tools/deployment/mmdet2torchserve.py -------------------------------------------------------------------------------- /mmdet/tools/deployment/mmdet_handler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/tools/deployment/mmdet_handler.py -------------------------------------------------------------------------------- /mmdet/tools/deployment/onnx2tensorrt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/tools/deployment/onnx2tensorrt.py -------------------------------------------------------------------------------- /mmdet/tools/deployment/pytorch2onnx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/tools/deployment/pytorch2onnx.py -------------------------------------------------------------------------------- /mmdet/tools/deployment/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/tools/deployment/test.py -------------------------------------------------------------------------------- /mmdet/tools/deployment/test_torchserver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/tools/deployment/test_torchserver.py -------------------------------------------------------------------------------- /mmdet/tools/dist_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/tools/dist_test.sh -------------------------------------------------------------------------------- /mmdet/tools/dist_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/tools/dist_train.sh -------------------------------------------------------------------------------- /mmdet/tools/misc/browse_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/tools/misc/browse_dataset.py -------------------------------------------------------------------------------- /mmdet/tools/misc/download_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/tools/misc/download_dataset.py -------------------------------------------------------------------------------- /mmdet/tools/misc/gen_coco_panoptic_test_info.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/tools/misc/gen_coco_panoptic_test_info.py -------------------------------------------------------------------------------- /mmdet/tools/misc/get_image_metas.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/tools/misc/get_image_metas.py -------------------------------------------------------------------------------- /mmdet/tools/misc/print_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/tools/misc/print_config.py -------------------------------------------------------------------------------- /mmdet/tools/misc/split_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/tools/misc/split_coco.py -------------------------------------------------------------------------------- /mmdet/tools/model_converters/detectron2pytorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/tools/model_converters/detectron2pytorch.py -------------------------------------------------------------------------------- /mmdet/tools/model_converters/publish_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/tools/model_converters/publish_model.py -------------------------------------------------------------------------------- /mmdet/tools/model_converters/regnet2mmdet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/tools/model_converters/regnet2mmdet.py -------------------------------------------------------------------------------- /mmdet/tools/model_converters/selfsup2mmdet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/tools/model_converters/selfsup2mmdet.py -------------------------------------------------------------------------------- /mmdet/tools/model_converters/upgrade_model_version.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/tools/model_converters/upgrade_model_version.py -------------------------------------------------------------------------------- /mmdet/tools/model_converters/upgrade_ssd_version.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/tools/model_converters/upgrade_ssd_version.py -------------------------------------------------------------------------------- /mmdet/tools/slurm_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/tools/slurm_test.sh -------------------------------------------------------------------------------- /mmdet/tools/slurm_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/tools/slurm_train.sh -------------------------------------------------------------------------------- /mmdet/tools/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/tools/test.py -------------------------------------------------------------------------------- /mmdet/tools/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/mmdet/tools/train.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/mal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/models/mal.py -------------------------------------------------------------------------------- /models/vision_transformers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/models/vision_transformers.py -------------------------------------------------------------------------------- /tools/convert2lmdb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/tools/convert2lmdb.py -------------------------------------------------------------------------------- /tools/move_objects365_data.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/tools/move_objects365_data.sh -------------------------------------------------------------------------------- /tools/mv_data_from_ddn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/tools/mv_data_from_ddn.py -------------------------------------------------------------------------------- /tools/visualize_anns.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/tools/visualize_anns.py -------------------------------------------------------------------------------- /tools/visualize_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/tools/visualize_attention.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/optimizers/adamw.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/mask-auto-labeler/HEAD/utils/optimizers/adamw.py --------------------------------------------------------------------------------