├── .gitignore ├── README.md ├── build.sh ├── configs ├── _base_ │ ├── datasets │ │ ├── coco_instance.py │ │ ├── kitti-3d-3class.py │ │ ├── kitti-3d-car.py │ │ ├── kitti-mono3d.py │ │ ├── lyft-3d.py │ │ ├── nuim_instance.py │ │ ├── nus-3d.py │ │ ├── nus-mono3d.py │ │ ├── range100_lyft-3d.py │ │ ├── s3dis-3d-5class.py │ │ ├── s3dis_seg-3d-13class.py │ │ ├── scannet-3d-18class.py │ │ ├── scannet_seg-3d-20class.py │ │ ├── sunrgbd-3d-10class.py │ │ ├── waymoD5-3d-3class.py │ │ └── waymoD5-3d-car.py │ ├── default_runtime.py │ ├── models │ │ ├── 3dssd.py │ │ ├── cascade_mask_rcnn_r50_fpn.py │ │ ├── centerpoint_01voxel_second_secfpn_nus.py │ │ ├── centerpoint_02pillar_second_secfpn_nus.py │ │ ├── dgcnn.py │ │ ├── fcos3d.py │ │ ├── groupfree3d.py │ │ ├── h3dnet.py │ │ ├── hv_pointpillars_fpn_lyft.py │ │ ├── hv_pointpillars_fpn_nus.py │ │ ├── hv_pointpillars_fpn_range100_lyft.py │ │ ├── hv_pointpillars_secfpn_kitti.py │ │ ├── hv_pointpillars_secfpn_waymo.py │ │ ├── hv_second_secfpn_kitti.py │ │ ├── hv_second_secfpn_waymo.py │ │ ├── imvotenet_image.py │ │ ├── mask_rcnn_r50_fpn.py │ │ ├── paconv_cuda_ssg.py │ │ ├── paconv_ssg.py │ │ ├── parta2.py │ │ ├── pgd.py │ │ ├── point_rcnn.py │ │ ├── pointnet2_msg.py │ │ ├── pointnet2_ssg.py │ │ ├── smoke.py │ │ └── votenet.py │ └── schedules │ │ ├── cosine.py │ │ ├── cyclic_20e.py │ │ ├── cyclic_40e.py │ │ ├── mmdet_schedule_1x.py │ │ ├── schedule_2x.py │ │ ├── schedule_3x.py │ │ ├── seg_cosine_100e.py │ │ ├── seg_cosine_150e.py │ │ ├── seg_cosine_200e.py │ │ └── seg_cosine_50e.py ├── dort │ ├── dort_bevdepth_resnet50_768x256-bsz4.py │ └── dort_bevdepth_resnet50_768x256.py └── tracking_configs │ └── sort │ ├── ab3dmot_bevdet_tiny_cbgs_res1056x384.py │ └── ab3dmot_imvoxelnet_nusc.py ├── det3d ├── __init__.py ├── apis │ ├── __init__.py │ └── test.py ├── core │ ├── bbox │ │ ├── assigners │ │ │ ├── __init__.py │ │ │ └── hungarian_assigner_3d.py │ │ ├── coders │ │ │ ├── __init__.py │ │ │ ├── nms_free_coder.py │ │ │ └── nocs_coder.py │ │ ├── match_costs │ │ │ ├── __init__.py │ │ │ └── match_cost.py │ │ └── util.py │ └── visualizer │ │ ├── __init__.py │ │ ├── show_result.py │ │ └── utils.py ├── datasets │ ├── __init__.py │ ├── kitti_dataset.py │ ├── nuscenes_bevdet_dataset.py │ ├── nuscenes_dataset.py │ ├── nuscenes_frame_dataset.py │ ├── nuscenes_mono_dataset.py │ ├── nuscenes_mono_frame_dataset.py │ ├── pipelines │ │ ├── __init__.py │ │ ├── bev_transform_3d.py │ │ ├── custom_transform_3d.py │ │ ├── formating.py │ │ └── transform_3d.py │ └── waymo_dataset.py ├── hooks │ ├── __init__.py │ ├── ema.py │ ├── mlflow.py │ └── sequentialcontrol.py ├── models │ ├── backbones │ │ ├── DCNv2 │ │ │ ├── .gitignore │ │ │ ├── LICENSE │ │ │ ├── README.md │ │ │ ├── __init__.py │ │ │ ├── dcn_v2.py │ │ │ ├── dcn_v2.py_double │ │ │ ├── dcn_v2_onnx.py │ │ │ ├── make.sh │ │ │ ├── setup.py │ │ │ ├── src │ │ │ │ ├── cpu │ │ │ │ │ ├── dcn_v2_cpu.cpp │ │ │ │ │ ├── dcn_v2_im2col_cpu.cpp │ │ │ │ │ ├── dcn_v2_im2col_cpu.h │ │ │ │ │ ├── dcn_v2_psroi_pooling_cpu.cpp │ │ │ │ │ └── vision.h │ │ │ │ ├── cuda │ │ │ │ │ ├── dcn_v2_cuda.cu │ │ │ │ │ ├── dcn_v2_im2col_cuda.cu │ │ │ │ │ ├── dcn_v2_im2col_cuda.h │ │ │ │ │ ├── dcn_v2_psroi_pooling_cuda.cu │ │ │ │ │ └── vision.h │ │ │ │ ├── dcn_v2.h │ │ │ │ └── vision.cpp │ │ │ ├── testcpu.py │ │ │ └── testcuda.py │ │ ├── DCNv2_t18 │ │ │ ├── LICENSE │ │ │ ├── README.md │ │ │ ├── __init__.py │ │ │ ├── dcn_v2.py │ │ │ ├── dcn_v2_onnx.py │ │ │ ├── make.sh │ │ │ ├── setup.py │ │ │ ├── src │ │ │ │ ├── cpu │ │ │ │ │ ├── dcn_v2_cpu.cpp │ │ │ │ │ ├── dcn_v2_im2col_cpu.cpp │ │ │ │ │ ├── dcn_v2_im2col_cpu.h │ │ │ │ │ ├── dcn_v2_psroi_pooling_cpu.cpp │ │ │ │ │ └── vision.h │ │ │ │ ├── cuda │ │ │ │ │ ├── dcn_v2_cuda.cu │ │ │ │ │ ├── dcn_v2_im2col_cuda.cu │ │ │ │ │ ├── dcn_v2_im2col_cuda.h │ │ │ │ │ ├── dcn_v2_psroi_pooling_cuda.cu │ │ │ │ │ └── vision.h │ │ │ │ ├── dcn_v2.h │ │ │ │ └── vision.cpp │ │ │ ├── testcpu.py │ │ │ └── testcuda.py │ │ ├── __init__.py │ │ ├── pose_dla.py │ │ ├── pose_resnet_dcn.py │ │ ├── resnet.py │ │ ├── swin.py │ │ ├── swin_utils.py │ │ └── vovnet.py │ ├── dense_heads │ │ ├── __init__.py │ │ ├── bev_object_head.py │ │ ├── centernet3d_head.py │ │ ├── centernet_utils.py │ │ ├── custom_centerpoint_head.py │ │ ├── custom_pgd.py │ │ ├── detr3d_head.py │ │ ├── dgcnn3d_head.py │ │ ├── liga_head.py │ │ ├── loss_utils.py │ │ ├── two_stage_2d_head.py │ │ └── two_stage_head.py │ ├── detectors │ │ ├── __init__.py │ │ ├── centernet3d.py │ │ ├── custom_imvoxelnet.py │ │ ├── custom_imvoxelnetv2.py │ │ ├── detr3d.py │ │ ├── fusion_imvoxelnet.py │ │ ├── imdynamic.py │ │ ├── liga_stereo.py │ │ ├── local_volume.py │ │ └── obj_dgcnn.py │ ├── fusion_layers │ │ └── point_fusion.py │ ├── losses │ │ ├── __init__.py │ │ └── uncertainty_l1_loss.py │ ├── motion │ │ ├── __init__.py │ │ └── ab3dmot_kalman_filter.py │ ├── necks │ │ ├── __init__.py │ │ ├── bevdepth_lift_splat.py │ │ ├── bevdet_fpn.py │ │ ├── fpn_lss_neck.py │ │ ├── fpn_lss_neckv2.py │ │ ├── imdynamic.py │ │ ├── imdynamic_lss.py │ │ ├── imvoxel_neck.py │ │ ├── imvoxel_view_transform.py │ │ ├── lift_splat.py │ │ ├── lift_splatv2.py │ │ ├── liga_neck.py │ │ ├── resnet_bevdet.py │ │ ├── second_fpn.py │ │ └── transform_grid_sample.py │ ├── track_frameworks │ │ ├── __init__.py │ │ ├── qdtrack.py │ │ └── sort.py │ ├── track_heads │ │ ├── __init__.py │ │ └── quasi_dense_track_head.py │ ├── trackers │ │ ├── __init__.py │ │ ├── quasi_dense_tracker.py │ │ ├── sort_tracker.py │ │ └── utils.py │ └── utils │ │ ├── __init__.py │ │ ├── detr.py │ │ ├── detr3d_transformer.py │ │ ├── dgcnn_attn.py │ │ ├── grid_mask.py │ │ └── noc_detr3d_transformer.py ├── ops │ ├── __init__.py │ ├── bev_pool_v2 │ │ ├── __init__.py │ │ ├── bev_pool.py │ │ └── src │ │ │ ├── bev_pool.cpp │ │ │ └── bev_pool_cuda.cu │ └── build_cost_volume │ │ ├── __init__.py │ │ └── src │ │ ├── BuildCostVolume.cpp │ │ └── BuildCostVolume_cuda.cu ├── utils │ ├── manager.py │ └── misc.py └── version.py ├── figs └── overview.png ├── setup.py ├── software ├── mmdet │ ├── .circleci │ │ └── config.yml │ ├── .dev_scripts │ │ ├── batch_test_list.py │ │ ├── batch_train_list.txt │ │ ├── benchmark_filter.py │ │ ├── benchmark_inference_fps.py │ │ ├── benchmark_test_image.py │ │ ├── check_links.py │ │ ├── convert_test_benchmark_script.py │ │ ├── convert_train_benchmark_script.py │ │ ├── gather_models.py │ │ ├── gather_test_benchmark_metric.py │ │ ├── gather_train_benchmark_metric.py │ │ ├── linter.sh │ │ ├── test_benchmark.sh │ │ ├── test_init_backbone.py │ │ └── train_benchmark.sh │ ├── .owners.yml │ ├── .pre-commit-config.yaml │ ├── .readthedocs.yml │ ├── CITATION.cff │ ├── LICENSE │ ├── MANIFEST.in │ ├── README.md │ ├── README_zh-CN.md │ ├── configs │ │ ├── _base_ │ │ │ ├── datasets │ │ │ │ ├── cityscapes_detection.py │ │ │ │ ├── cityscapes_instance.py │ │ │ │ ├── coco_detection.py │ │ │ │ ├── coco_instance.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 │ │ ├── albu_example │ │ │ ├── README.md │ │ │ └── mask_rcnn_r50_fpn_albu_1x_coco.py │ │ ├── atss │ │ │ ├── README.md │ │ │ ├── atss_r101_fpn_1x_coco.py │ │ │ ├── atss_r50_fpn_1x_coco.py │ │ │ └── metafile.yml │ │ ├── autoassign │ │ │ ├── README.md │ │ │ ├── autoassign_r50_fpn_8x2_1x_coco.py │ │ │ └── metafile.yml │ │ ├── carafe │ │ │ ├── README.md │ │ │ ├── faster_rcnn_r50_fpn_carafe_1x_coco.py │ │ │ ├── mask_rcnn_r50_fpn_carafe_1x_coco.py │ │ │ └── metafile.yml │ │ ├── cascade_rcnn │ │ │ ├── README.md │ │ │ ├── cascade_mask_rcnn_r101_caffe_fpn_1x_coco.py │ │ │ ├── cascade_mask_rcnn_r101_caffe_fpn_mstrain_3x_coco.py │ │ │ ├── cascade_mask_rcnn_r101_fpn_1x_coco.py │ │ │ ├── cascade_mask_rcnn_r101_fpn_20e_coco.py │ │ │ ├── cascade_mask_rcnn_r101_fpn_mstrain_3x_coco.py │ │ │ ├── cascade_mask_rcnn_r50_caffe_fpn_1x_coco.py │ │ │ ├── cascade_mask_rcnn_r50_caffe_fpn_mstrain_3x_coco.py │ │ │ ├── cascade_mask_rcnn_r50_fpn_1x_coco.py │ │ │ ├── cascade_mask_rcnn_r50_fpn_20e_coco.py │ │ │ ├── cascade_mask_rcnn_r50_fpn_mstrain_3x_coco.py │ │ │ ├── cascade_mask_rcnn_x101_32x4d_fpn_1x_coco.py │ │ │ ├── cascade_mask_rcnn_x101_32x4d_fpn_20e_coco.py │ │ │ ├── cascade_mask_rcnn_x101_32x4d_fpn_mstrain_3x_coco.py │ │ │ ├── cascade_mask_rcnn_x101_32x8d_fpn_mstrain_3x_coco.py │ │ │ ├── cascade_mask_rcnn_x101_64x4d_fpn_1x_coco.py │ │ │ ├── cascade_mask_rcnn_x101_64x4d_fpn_20e_coco.py │ │ │ ├── cascade_mask_rcnn_x101_64x4d_fpn_mstrain_3x_coco.py │ │ │ ├── cascade_rcnn_r101_caffe_fpn_1x_coco.py │ │ │ ├── cascade_rcnn_r101_fpn_1x_coco.py │ │ │ ├── cascade_rcnn_r101_fpn_20e_coco.py │ │ │ ├── cascade_rcnn_r50_caffe_fpn_1x_coco.py │ │ │ ├── cascade_rcnn_r50_fpn_1x_coco.py │ │ │ ├── cascade_rcnn_r50_fpn_20e_coco.py │ │ │ ├── cascade_rcnn_x101_32x4d_fpn_1x_coco.py │ │ │ ├── cascade_rcnn_x101_32x4d_fpn_20e_coco.py │ │ │ ├── cascade_rcnn_x101_64x4d_fpn_1x_coco.py │ │ │ ├── cascade_rcnn_x101_64x4d_fpn_20e_coco.py │ │ │ └── metafile.yml │ │ ├── cascade_rpn │ │ │ ├── README.md │ │ │ ├── crpn_fast_rcnn_r50_caffe_fpn_1x_coco.py │ │ │ ├── crpn_faster_rcnn_r50_caffe_fpn_1x_coco.py │ │ │ ├── crpn_r50_caffe_fpn_1x_coco.py │ │ │ └── metafile.yml │ │ ├── centernet │ │ │ ├── README.md │ │ │ ├── centernet_resnet18_140e_coco.py │ │ │ ├── centernet_resnet18_dcnv2_140e_coco.py │ │ │ └── metafile.yml │ │ ├── centripetalnet │ │ │ ├── README.md │ │ │ ├── centripetalnet_hourglass104_mstest_16x6_210e_coco.py │ │ │ └── metafile.yml │ │ ├── cityscapes │ │ │ ├── README.md │ │ │ ├── faster_rcnn_r50_fpn_1x_cityscapes.py │ │ │ └── mask_rcnn_r50_fpn_1x_cityscapes.py │ │ ├── common │ │ │ ├── lsj_100e_coco_instance.py │ │ │ ├── mstrain-poly_3x_coco_instance.py │ │ │ ├── mstrain_3x_coco.py │ │ │ ├── mstrain_3x_coco_instance.py │ │ │ ├── ssj_270k_coco_instance.py │ │ │ └── ssj_scp_270k_coco_instance.py │ │ ├── convnext │ │ │ ├── README.md │ │ │ ├── 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 │ │ │ ├── mask_rcnn_convnext-t_p4_w7_fpn_fp16_ms-crop_3x_coco.py │ │ │ └── metafile.yml │ │ ├── cornernet │ │ │ ├── README.md │ │ │ ├── cornernet_hourglass104_mstest_10x5_210e_coco.py │ │ │ ├── cornernet_hourglass104_mstest_32x3_210e_coco.py │ │ │ ├── cornernet_hourglass104_mstest_8x6_210e_coco.py │ │ │ └── metafile.yml │ │ ├── dcn │ │ │ ├── README.md │ │ │ ├── cascade_mask_rcnn_r101_fpn_dconv_c3-c5_1x_coco.py │ │ │ ├── cascade_mask_rcnn_r50_fpn_dconv_c3-c5_1x_coco.py │ │ │ ├── cascade_mask_rcnn_x101_32x4d_fpn_dconv_c3-c5_1x_coco.py │ │ │ ├── cascade_rcnn_r101_fpn_dconv_c3-c5_1x_coco.py │ │ │ ├── cascade_rcnn_r50_fpn_dconv_c3-c5_1x_coco.py │ │ │ ├── faster_rcnn_r101_fpn_dconv_c3-c5_1x_coco.py │ │ │ ├── faster_rcnn_r50_fpn_dconv_c3-c5_1x_coco.py │ │ │ ├── faster_rcnn_r50_fpn_dpool_1x_coco.py │ │ │ ├── faster_rcnn_x101_32x4d_fpn_dconv_c3-c5_1x_coco.py │ │ │ ├── mask_rcnn_r101_fpn_dconv_c3-c5_1x_coco.py │ │ │ ├── mask_rcnn_r50_fpn_dconv_c3-c5_1x_coco.py │ │ │ ├── mask_rcnn_r50_fpn_fp16_dconv_c3-c5_1x_coco.py │ │ │ └── metafile.yml │ │ ├── dcnv2 │ │ │ ├── README.md │ │ │ ├── faster_rcnn_r50_fpn_mdconv_c3-c5_1x_coco.py │ │ │ ├── faster_rcnn_r50_fpn_mdconv_c3-c5_group4_1x_coco.py │ │ │ ├── faster_rcnn_r50_fpn_mdpool_1x_coco.py │ │ │ ├── mask_rcnn_r50_fpn_fp16_mdconv_c3-c5_1x_coco.py │ │ │ ├── mask_rcnn_r50_fpn_mdconv_c3-c5_1x_coco.py │ │ │ └── metafile.yml │ │ ├── ddod │ │ │ ├── README.md │ │ │ ├── ddod_r50_fpn_1x_coco.py │ │ │ └── metafile.yml │ │ ├── deepfashion │ │ │ ├── README.md │ │ │ └── mask_rcnn_r50_fpn_15e_deepfashion.py │ │ ├── deformable_detr │ │ │ ├── README.md │ │ │ ├── deformable_detr_r50_16x2_50e_coco.py │ │ │ ├── deformable_detr_refine_r50_16x2_50e_coco.py │ │ │ ├── deformable_detr_twostage_refine_r50_16x2_50e_coco.py │ │ │ └── metafile.yml │ │ ├── detectors │ │ │ ├── README.md │ │ │ ├── cascade_rcnn_r50_rfp_1x_coco.py │ │ │ ├── cascade_rcnn_r50_sac_1x_coco.py │ │ │ ├── detectors_cascade_rcnn_r50_1x_coco.py │ │ │ ├── detectors_htc_r101_20e_coco.py │ │ │ ├── detectors_htc_r50_1x_coco.py │ │ │ ├── htc_r50_rfp_1x_coco.py │ │ │ ├── htc_r50_sac_1x_coco.py │ │ │ └── metafile.yml │ │ ├── detr │ │ │ ├── README.md │ │ │ ├── detr_r50_8x2_150e_coco.py │ │ │ └── metafile.yml │ │ ├── double_heads │ │ │ ├── README.md │ │ │ ├── dh_faster_rcnn_r50_fpn_1x_coco.py │ │ │ └── metafile.yml │ │ ├── dyhead │ │ │ ├── README.md │ │ │ ├── atss_r50_caffe_fpn_dyhead_1x_coco.py │ │ │ ├── atss_r50_fpn_dyhead_1x_coco.py │ │ │ ├── atss_swin-l-p4-w12_fpn_dyhead_mstrain_2x_coco.py │ │ │ └── metafile.yml │ │ ├── dynamic_rcnn │ │ │ ├── README.md │ │ │ ├── dynamic_rcnn_r50_fpn_1x_coco.py │ │ │ └── metafile.yml │ │ ├── efficientnet │ │ │ ├── README.md │ │ │ ├── metafile.yml │ │ │ └── retinanet_effb3_fpn_crop896_8x4_1x_coco.py │ │ ├── empirical_attention │ │ │ ├── README.md │ │ │ ├── faster_rcnn_r50_fpn_attention_0010_1x_coco.py │ │ │ ├── faster_rcnn_r50_fpn_attention_0010_dcn_1x_coco.py │ │ │ ├── faster_rcnn_r50_fpn_attention_1111_1x_coco.py │ │ │ ├── faster_rcnn_r50_fpn_attention_1111_dcn_1x_coco.py │ │ │ └── metafile.yml │ │ ├── fast_rcnn │ │ │ ├── README.md │ │ │ ├── fast_rcnn_r101_caffe_fpn_1x_coco.py │ │ │ ├── fast_rcnn_r101_fpn_1x_coco.py │ │ │ ├── fast_rcnn_r101_fpn_2x_coco.py │ │ │ ├── fast_rcnn_r50_caffe_fpn_1x_coco.py │ │ │ ├── fast_rcnn_r50_fpn_1x_coco.py │ │ │ └── fast_rcnn_r50_fpn_2x_coco.py │ │ ├── faster_rcnn │ │ │ ├── README.md │ │ │ ├── faster_rcnn_r101_caffe_fpn_1x_coco.py │ │ │ ├── faster_rcnn_r101_caffe_fpn_mstrain_3x_coco.py │ │ │ ├── faster_rcnn_r101_fpn_1x_coco.py │ │ │ ├── faster_rcnn_r101_fpn_2x_coco.py │ │ │ ├── faster_rcnn_r101_fpn_mstrain_3x_coco.py │ │ │ ├── faster_rcnn_r50_caffe_c4_1x_coco.py │ │ │ ├── faster_rcnn_r50_caffe_c4_mstrain_1x_coco.py │ │ │ ├── faster_rcnn_r50_caffe_dc5_1x_coco.py │ │ │ ├── faster_rcnn_r50_caffe_dc5_mstrain_1x_coco.py │ │ │ ├── faster_rcnn_r50_caffe_dc5_mstrain_3x_coco.py │ │ │ ├── faster_rcnn_r50_caffe_fpn_1x_coco.py │ │ │ ├── faster_rcnn_r50_caffe_fpn_90k_coco.py │ │ │ ├── faster_rcnn_r50_caffe_fpn_mstrain_1x_coco-person-bicycle-car.py │ │ │ ├── faster_rcnn_r50_caffe_fpn_mstrain_1x_coco-person.py │ │ │ ├── faster_rcnn_r50_caffe_fpn_mstrain_1x_coco.py │ │ │ ├── faster_rcnn_r50_caffe_fpn_mstrain_2x_coco.py │ │ │ ├── faster_rcnn_r50_caffe_fpn_mstrain_3x_coco.py │ │ │ ├── faster_rcnn_r50_caffe_fpn_mstrain_90k_coco.py │ │ │ ├── faster_rcnn_r50_fpn_1x_coco.py │ │ │ ├── faster_rcnn_r50_fpn_2x_coco.py │ │ │ ├── faster_rcnn_r50_fpn_bounded_iou_1x_coco.py │ │ │ ├── faster_rcnn_r50_fpn_ciou_1x_coco.py │ │ │ ├── faster_rcnn_r50_fpn_fp16_1x_coco.py │ │ │ ├── faster_rcnn_r50_fpn_giou_1x_coco.py │ │ │ ├── faster_rcnn_r50_fpn_iou_1x_coco.py │ │ │ ├── faster_rcnn_r50_fpn_mstrain_3x_coco.py │ │ │ ├── faster_rcnn_r50_fpn_ohem_1x_coco.py │ │ │ ├── faster_rcnn_r50_fpn_soft_nms_1x_coco.py │ │ │ ├── faster_rcnn_r50_fpn_tnr-pretrain_1x_coco.py │ │ │ ├── faster_rcnn_x101_32x4d_fpn_1x_coco.py │ │ │ ├── faster_rcnn_x101_32x4d_fpn_2x_coco.py │ │ │ ├── faster_rcnn_x101_32x4d_fpn_mstrain_3x_coco.py │ │ │ ├── faster_rcnn_x101_32x8d_fpn_mstrain_3x_coco.py │ │ │ ├── faster_rcnn_x101_64x4d_fpn_1x_coco.py │ │ │ ├── faster_rcnn_x101_64x4d_fpn_2x_coco.py │ │ │ ├── faster_rcnn_x101_64x4d_fpn_mstrain_3x_coco.py │ │ │ └── metafile.yml │ │ ├── fcos │ │ │ ├── README.md │ │ │ ├── fcos_center-normbbox-centeronreg-giou_r50_caffe_fpn_gn-head_1x_coco.py │ │ │ ├── fcos_center-normbbox-centeronreg-giou_r50_caffe_fpn_gn-head_dcn_1x_coco.py │ │ │ ├── fcos_center_r50_caffe_fpn_gn-head_1x_coco.py │ │ │ ├── fcos_r101_caffe_fpn_gn-head_1x_coco.py │ │ │ ├── fcos_r101_caffe_fpn_gn-head_mstrain_640-800_2x_coco.py │ │ │ ├── fcos_r50_caffe_fpn_gn-head_1x_coco.py │ │ │ ├── fcos_r50_caffe_fpn_gn-head_4x4_1x_coco.py │ │ │ ├── fcos_r50_caffe_fpn_gn-head_mstrain_640-800_2x_coco.py │ │ │ ├── fcos_x101_64x4d_fpn_gn-head_mstrain_640-800_2x_coco.py │ │ │ └── metafile.yml │ │ ├── foveabox │ │ │ ├── README.md │ │ │ ├── fovea_align_r101_fpn_gn-head_4x4_2x_coco.py │ │ │ ├── fovea_align_r101_fpn_gn-head_mstrain_640-800_4x4_2x_coco.py │ │ │ ├── fovea_align_r50_fpn_gn-head_4x4_2x_coco.py │ │ │ ├── fovea_align_r50_fpn_gn-head_mstrain_640-800_4x4_2x_coco.py │ │ │ ├── fovea_r101_fpn_4x4_1x_coco.py │ │ │ ├── fovea_r101_fpn_4x4_2x_coco.py │ │ │ ├── fovea_r50_fpn_4x4_1x_coco.py │ │ │ ├── fovea_r50_fpn_4x4_2x_coco.py │ │ │ └── metafile.yml │ │ ├── fpg │ │ │ ├── README.md │ │ │ ├── faster_rcnn_r50_fpg-chn128_crop640_50e_coco.py │ │ │ ├── faster_rcnn_r50_fpg_crop640_50e_coco.py │ │ │ ├── faster_rcnn_r50_fpn_crop640_50e_coco.py │ │ │ ├── mask_rcnn_r50_fpg-chn128_crop640_50e_coco.py │ │ │ ├── mask_rcnn_r50_fpg_crop640_50e_coco.py │ │ │ ├── mask_rcnn_r50_fpn_crop640_50e_coco.py │ │ │ ├── metafile.yml │ │ │ ├── retinanet_r50_fpg-chn128_crop640_50e_coco.py │ │ │ └── retinanet_r50_fpg_crop640_50e_coco.py │ │ ├── free_anchor │ │ │ ├── README.md │ │ │ ├── metafile.yml │ │ │ ├── retinanet_free_anchor_r101_fpn_1x_coco.py │ │ │ ├── retinanet_free_anchor_r50_fpn_1x_coco.py │ │ │ └── retinanet_free_anchor_x101_32x4d_fpn_1x_coco.py │ │ ├── fsaf │ │ │ ├── README.md │ │ │ ├── fsaf_r101_fpn_1x_coco.py │ │ │ ├── fsaf_r50_fpn_1x_coco.py │ │ │ ├── fsaf_x101_64x4d_fpn_1x_coco.py │ │ │ └── metafile.yml │ │ ├── gcnet │ │ │ ├── README.md │ │ │ ├── cascade_mask_rcnn_x101_32x4d_fpn_syncbn-backbone_1x_coco.py │ │ │ ├── cascade_mask_rcnn_x101_32x4d_fpn_syncbn-backbone_dconv_c3-c5_1x_coco.py │ │ │ ├── cascade_mask_rcnn_x101_32x4d_fpn_syncbn-backbone_dconv_c3-c5_r16_gcb_c3-c5_1x_coco.py │ │ │ ├── cascade_mask_rcnn_x101_32x4d_fpn_syncbn-backbone_dconv_c3-c5_r4_gcb_c3-c5_1x_coco.py │ │ │ ├── cascade_mask_rcnn_x101_32x4d_fpn_syncbn-backbone_r16_gcb_c3-c5_1x_coco.py │ │ │ ├── cascade_mask_rcnn_x101_32x4d_fpn_syncbn-backbone_r4_gcb_c3-c5_1x_coco.py │ │ │ ├── mask_rcnn_r101_fpn_r16_gcb_c3-c5_1x_coco.py │ │ │ ├── mask_rcnn_r101_fpn_r4_gcb_c3-c5_1x_coco.py │ │ │ ├── mask_rcnn_r101_fpn_syncbn-backbone_1x_coco.py │ │ │ ├── mask_rcnn_r101_fpn_syncbn-backbone_r16_gcb_c3-c5_1x_coco.py │ │ │ ├── mask_rcnn_r101_fpn_syncbn-backbone_r4_gcb_c3-c5_1x_coco.py │ │ │ ├── mask_rcnn_r50_fpn_r16_gcb_c3-c5_1x_coco.py │ │ │ ├── mask_rcnn_r50_fpn_r4_gcb_c3-c5_1x_coco.py │ │ │ ├── mask_rcnn_r50_fpn_syncbn-backbone_1x_coco.py │ │ │ ├── mask_rcnn_r50_fpn_syncbn-backbone_r16_gcb_c3-c5_1x_coco.py │ │ │ ├── mask_rcnn_r50_fpn_syncbn-backbone_r4_gcb_c3-c5_1x_coco.py │ │ │ ├── mask_rcnn_x101_32x4d_fpn_syncbn-backbone_1x_coco.py │ │ │ ├── mask_rcnn_x101_32x4d_fpn_syncbn-backbone_r16_gcb_c3-c5_1x_coco.py │ │ │ ├── mask_rcnn_x101_32x4d_fpn_syncbn-backbone_r4_gcb_c3-c5_1x_coco.py │ │ │ └── metafile.yml │ │ ├── gfl │ │ │ ├── README.md │ │ │ ├── gfl_r101_fpn_dconv_c3-c5_mstrain_2x_coco.py │ │ │ ├── gfl_r101_fpn_mstrain_2x_coco.py │ │ │ ├── gfl_r50_fpn_1x_coco.py │ │ │ ├── gfl_r50_fpn_mstrain_2x_coco.py │ │ │ ├── gfl_x101_32x4d_fpn_dconv_c4-c5_mstrain_2x_coco.py │ │ │ ├── gfl_x101_32x4d_fpn_mstrain_2x_coco.py │ │ │ └── metafile.yml │ │ ├── ghm │ │ │ ├── README.md │ │ │ ├── metafile.yml │ │ │ ├── retinanet_ghm_r101_fpn_1x_coco.py │ │ │ ├── retinanet_ghm_r50_fpn_1x_coco.py │ │ │ ├── retinanet_ghm_x101_32x4d_fpn_1x_coco.py │ │ │ └── retinanet_ghm_x101_64x4d_fpn_1x_coco.py │ │ ├── gn+ws │ │ │ ├── README.md │ │ │ ├── faster_rcnn_r101_fpn_gn_ws-all_1x_coco.py │ │ │ ├── faster_rcnn_r50_fpn_gn_ws-all_1x_coco.py │ │ │ ├── faster_rcnn_x101_32x4d_fpn_gn_ws-all_1x_coco.py │ │ │ ├── faster_rcnn_x50_32x4d_fpn_gn_ws-all_1x_coco.py │ │ │ ├── mask_rcnn_r101_fpn_gn_ws-all_20_23_24e_coco.py │ │ │ ├── mask_rcnn_r101_fpn_gn_ws-all_2x_coco.py │ │ │ ├── mask_rcnn_r50_fpn_gn_ws-all_20_23_24e_coco.py │ │ │ ├── mask_rcnn_r50_fpn_gn_ws-all_2x_coco.py │ │ │ ├── mask_rcnn_x101_32x4d_fpn_gn_ws-all_20_23_24e_coco.py │ │ │ ├── mask_rcnn_x101_32x4d_fpn_gn_ws-all_2x_coco.py │ │ │ ├── mask_rcnn_x50_32x4d_fpn_gn_ws-all_20_23_24e_coco.py │ │ │ ├── mask_rcnn_x50_32x4d_fpn_gn_ws-all_2x_coco.py │ │ │ └── metafile.yml │ │ ├── gn │ │ │ ├── README.md │ │ │ ├── mask_rcnn_r101_fpn_gn-all_2x_coco.py │ │ │ ├── mask_rcnn_r101_fpn_gn-all_3x_coco.py │ │ │ ├── mask_rcnn_r50_fpn_gn-all_2x_coco.py │ │ │ ├── mask_rcnn_r50_fpn_gn-all_3x_coco.py │ │ │ ├── mask_rcnn_r50_fpn_gn-all_contrib_2x_coco.py │ │ │ ├── mask_rcnn_r50_fpn_gn-all_contrib_3x_coco.py │ │ │ └── metafile.yml │ │ ├── grid_rcnn │ │ │ ├── README.md │ │ │ ├── grid_rcnn_r101_fpn_gn-head_2x_coco.py │ │ │ ├── grid_rcnn_r50_fpn_gn-head_1x_coco.py │ │ │ ├── grid_rcnn_r50_fpn_gn-head_2x_coco.py │ │ │ ├── grid_rcnn_x101_32x4d_fpn_gn-head_2x_coco.py │ │ │ ├── grid_rcnn_x101_64x4d_fpn_gn-head_2x_coco.py │ │ │ └── metafile.yml │ │ ├── groie │ │ │ ├── README.md │ │ │ ├── faster_rcnn_r50_fpn_groie_1x_coco.py │ │ │ ├── grid_rcnn_r50_fpn_gn-head_groie_1x_coco.py │ │ │ ├── mask_rcnn_r101_fpn_syncbn-backbone_r4_gcb_c3-c5_groie_1x_coco.py │ │ │ ├── mask_rcnn_r50_fpn_groie_1x_coco.py │ │ │ ├── mask_rcnn_r50_fpn_syncbn-backbone_r4_gcb_c3-c5_groie_1x_coco.py │ │ │ └── metafile.yml │ │ ├── guided_anchoring │ │ │ ├── README.md │ │ │ ├── ga_fast_r50_caffe_fpn_1x_coco.py │ │ │ ├── ga_faster_r101_caffe_fpn_1x_coco.py │ │ │ ├── ga_faster_r50_caffe_fpn_1x_coco.py │ │ │ ├── ga_faster_r50_fpn_1x_coco.py │ │ │ ├── ga_faster_x101_32x4d_fpn_1x_coco.py │ │ │ ├── ga_faster_x101_64x4d_fpn_1x_coco.py │ │ │ ├── ga_retinanet_r101_caffe_fpn_1x_coco.py │ │ │ ├── ga_retinanet_r101_caffe_fpn_mstrain_2x.py │ │ │ ├── ga_retinanet_r50_caffe_fpn_1x_coco.py │ │ │ ├── ga_retinanet_r50_fpn_1x_coco.py │ │ │ ├── ga_retinanet_x101_32x4d_fpn_1x_coco.py │ │ │ ├── ga_retinanet_x101_64x4d_fpn_1x_coco.py │ │ │ ├── ga_rpn_r101_caffe_fpn_1x_coco.py │ │ │ ├── ga_rpn_r50_caffe_fpn_1x_coco.py │ │ │ ├── ga_rpn_r50_fpn_1x_coco.py │ │ │ ├── ga_rpn_x101_32x4d_fpn_1x_coco.py │ │ │ ├── ga_rpn_x101_64x4d_fpn_1x_coco.py │ │ │ └── metafile.yml │ │ ├── hrnet │ │ │ ├── README.md │ │ │ ├── cascade_mask_rcnn_hrnetv2p_w18_20e_coco.py │ │ │ ├── cascade_mask_rcnn_hrnetv2p_w32_20e_coco.py │ │ │ ├── cascade_mask_rcnn_hrnetv2p_w40_20e_coco.py │ │ │ ├── cascade_rcnn_hrnetv2p_w18_20e_coco.py │ │ │ ├── cascade_rcnn_hrnetv2p_w32_20e_coco.py │ │ │ ├── cascade_rcnn_hrnetv2p_w40_20e_coco.py │ │ │ ├── faster_rcnn_hrnetv2p_w18_1x_coco.py │ │ │ ├── faster_rcnn_hrnetv2p_w18_2x_coco.py │ │ │ ├── faster_rcnn_hrnetv2p_w32_1x_coco.py │ │ │ ├── faster_rcnn_hrnetv2p_w32_2x_coco.py │ │ │ ├── faster_rcnn_hrnetv2p_w40_1x_coco.py │ │ │ ├── faster_rcnn_hrnetv2p_w40_2x_coco.py │ │ │ ├── fcos_hrnetv2p_w18_gn-head_4x4_1x_coco.py │ │ │ ├── fcos_hrnetv2p_w18_gn-head_4x4_2x_coco.py │ │ │ ├── fcos_hrnetv2p_w18_gn-head_mstrain_640-800_4x4_2x_coco.py │ │ │ ├── fcos_hrnetv2p_w32_gn-head_4x4_1x_coco.py │ │ │ ├── fcos_hrnetv2p_w32_gn-head_4x4_2x_coco.py │ │ │ ├── fcos_hrnetv2p_w32_gn-head_mstrain_640-800_4x4_2x_coco.py │ │ │ ├── fcos_hrnetv2p_w40_gn-head_mstrain_640-800_4x4_2x_coco.py │ │ │ ├── htc_hrnetv2p_w18_20e_coco.py │ │ │ ├── htc_hrnetv2p_w32_20e_coco.py │ │ │ ├── htc_hrnetv2p_w40_20e_coco.py │ │ │ ├── htc_hrnetv2p_w40_28e_coco.py │ │ │ ├── htc_x101_64x4d_fpn_16x1_28e_coco.py │ │ │ ├── mask_rcnn_hrnetv2p_w18_1x_coco.py │ │ │ ├── mask_rcnn_hrnetv2p_w18_2x_coco.py │ │ │ ├── mask_rcnn_hrnetv2p_w32_1x_coco.py │ │ │ ├── mask_rcnn_hrnetv2p_w32_2x_coco.py │ │ │ ├── mask_rcnn_hrnetv2p_w40_1x_coco.py │ │ │ ├── mask_rcnn_hrnetv2p_w40_2x_coco.py │ │ │ └── metafile.yml │ │ ├── htc │ │ │ ├── README.md │ │ │ ├── htc_r101_fpn_20e_coco.py │ │ │ ├── htc_r50_fpn_1x_coco.py │ │ │ ├── htc_r50_fpn_20e_coco.py │ │ │ ├── htc_without_semantic_r50_fpn_1x_coco.py │ │ │ ├── htc_x101_32x4d_fpn_16x1_20e_coco.py │ │ │ ├── htc_x101_64x4d_fpn_16x1_20e_coco.py │ │ │ ├── htc_x101_64x4d_fpn_dconv_c3-c5_mstrain_400_1400_16x1_20e_coco.py │ │ │ └── metafile.yml │ │ ├── instaboost │ │ │ ├── README.md │ │ │ ├── cascade_mask_rcnn_r101_fpn_instaboost_4x_coco.py │ │ │ ├── cascade_mask_rcnn_r50_fpn_instaboost_4x_coco.py │ │ │ ├── cascade_mask_rcnn_x101_64x4d_fpn_instaboost_4x_coco.py │ │ │ ├── mask_rcnn_r101_fpn_instaboost_4x_coco.py │ │ │ ├── mask_rcnn_r50_fpn_instaboost_4x_coco.py │ │ │ ├── mask_rcnn_x101_64x4d_fpn_instaboost_4x_coco.py │ │ │ └── metafile.yml │ │ ├── lad │ │ │ ├── README.md │ │ │ ├── lad_r101_paa_r50_fpn_coco_1x.py │ │ │ ├── lad_r50_paa_r101_fpn_coco_1x.py │ │ │ └── metafile.yml │ │ ├── ld │ │ │ ├── README.md │ │ │ ├── ld_r101_gflv1_r101dcn_fpn_coco_2x.py │ │ │ ├── ld_r18_gflv1_r101_fpn_coco_1x.py │ │ │ ├── ld_r34_gflv1_r101_fpn_coco_1x.py │ │ │ ├── ld_r50_gflv1_r101_fpn_coco_1x.py │ │ │ └── metafile.yml │ │ ├── legacy_1.x │ │ │ ├── README.md │ │ │ ├── cascade_mask_rcnn_r50_fpn_1x_coco_v1.py │ │ │ ├── faster_rcnn_r50_fpn_1x_coco_v1.py │ │ │ ├── mask_rcnn_r50_fpn_1x_coco_v1.py │ │ │ ├── retinanet_r50_caffe_fpn_1x_coco_v1.py │ │ │ ├── retinanet_r50_fpn_1x_coco_v1.py │ │ │ └── ssd300_coco_v1.py │ │ ├── libra_rcnn │ │ │ ├── README.md │ │ │ ├── libra_fast_rcnn_r50_fpn_1x_coco.py │ │ │ ├── libra_faster_rcnn_r101_fpn_1x_coco.py │ │ │ ├── libra_faster_rcnn_r50_fpn_1x_coco.py │ │ │ ├── libra_faster_rcnn_x101_64x4d_fpn_1x_coco.py │ │ │ ├── libra_retinanet_r50_fpn_1x_coco.py │ │ │ └── metafile.yml │ │ ├── lvis │ │ │ ├── README.md │ │ │ ├── mask_rcnn_r101_fpn_sample1e-3_mstrain_1x_lvis_v1.py │ │ │ ├── mask_rcnn_r101_fpn_sample1e-3_mstrain_2x_lvis_v0.5.py │ │ │ ├── mask_rcnn_r50_fpn_sample1e-3_mstrain_1x_lvis_v1.py │ │ │ ├── mask_rcnn_r50_fpn_sample1e-3_mstrain_2x_lvis_v0.5.py │ │ │ ├── mask_rcnn_x101_32x4d_fpn_sample1e-3_mstrain_1x_lvis_v1.py │ │ │ ├── mask_rcnn_x101_32x4d_fpn_sample1e-3_mstrain_2x_lvis_v0.5.py │ │ │ ├── mask_rcnn_x101_64x4d_fpn_sample1e-3_mstrain_1x_lvis_v1.py │ │ │ └── mask_rcnn_x101_64x4d_fpn_sample1e-3_mstrain_2x_lvis_v0.5.py │ │ ├── mask2former │ │ │ ├── README.md │ │ │ ├── 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 │ │ │ └── metafile.yml │ │ ├── mask_rcnn │ │ │ ├── README.md │ │ │ ├── mask_rcnn_r101_caffe_fpn_1x_coco.py │ │ │ ├── mask_rcnn_r101_caffe_fpn_mstrain-poly_3x_coco.py │ │ │ ├── mask_rcnn_r101_fpn_1x_coco.py │ │ │ ├── mask_rcnn_r101_fpn_2x_coco.py │ │ │ ├── mask_rcnn_r101_fpn_mstrain-poly_3x_coco.py │ │ │ ├── mask_rcnn_r50_caffe_c4_1x_coco.py │ │ │ ├── mask_rcnn_r50_caffe_fpn_1x_coco.py │ │ │ ├── mask_rcnn_r50_caffe_fpn_mstrain-poly_1x_coco.py │ │ │ ├── mask_rcnn_r50_caffe_fpn_mstrain-poly_2x_coco.py │ │ │ ├── mask_rcnn_r50_caffe_fpn_mstrain-poly_3x_coco.py │ │ │ ├── mask_rcnn_r50_caffe_fpn_mstrain_1x_coco.py │ │ │ ├── mask_rcnn_r50_caffe_fpn_poly_1x_coco_v1.py │ │ │ ├── mask_rcnn_r50_fpn_1x_coco.py │ │ │ ├── mask_rcnn_r50_fpn_1x_wandb_coco.py │ │ │ ├── mask_rcnn_r50_fpn_2x_coco.py │ │ │ ├── mask_rcnn_r50_fpn_fp16_1x_coco.py │ │ │ ├── mask_rcnn_r50_fpn_mstrain-poly_3x_coco.py │ │ │ ├── mask_rcnn_r50_fpn_poly_1x_coco.py │ │ │ ├── mask_rcnn_x101_32x4d_fpn_1x_coco.py │ │ │ ├── mask_rcnn_x101_32x4d_fpn_2x_coco.py │ │ │ ├── mask_rcnn_x101_32x4d_fpn_mstrain-poly_3x_coco.py │ │ │ ├── mask_rcnn_x101_32x8d_fpn_1x_coco.py │ │ │ ├── mask_rcnn_x101_32x8d_fpn_mstrain-poly_1x_coco.py │ │ │ ├── mask_rcnn_x101_32x8d_fpn_mstrain-poly_3x_coco.py │ │ │ ├── mask_rcnn_x101_64x4d_fpn_1x_coco.py │ │ │ ├── mask_rcnn_x101_64x4d_fpn_2x_coco.py │ │ │ ├── mask_rcnn_x101_64x4d_fpn_mstrain-poly_3x_coco.py │ │ │ └── metafile.yml │ │ ├── maskformer │ │ │ ├── README.md │ │ │ ├── maskformer_r50_mstrain_16x1_75e_coco.py │ │ │ ├── maskformer_swin-l-p4-w12_mstrain_64x1_300e_coco.py │ │ │ └── metafile.yml │ │ ├── ms_rcnn │ │ │ ├── README.md │ │ │ ├── metafile.yml │ │ │ ├── ms_rcnn_r101_caffe_fpn_1x_coco.py │ │ │ ├── ms_rcnn_r101_caffe_fpn_2x_coco.py │ │ │ ├── ms_rcnn_r50_caffe_fpn_1x_coco.py │ │ │ ├── ms_rcnn_r50_caffe_fpn_2x_coco.py │ │ │ ├── ms_rcnn_r50_fpn_1x_coco.py │ │ │ ├── ms_rcnn_x101_32x4d_fpn_1x_coco.py │ │ │ ├── ms_rcnn_x101_64x4d_fpn_1x_coco.py │ │ │ └── ms_rcnn_x101_64x4d_fpn_2x_coco.py │ │ ├── nas_fcos │ │ │ ├── README.md │ │ │ ├── metafile.yml │ │ │ ├── nas_fcos_fcoshead_r50_caffe_fpn_gn-head_4x4_1x_coco.py │ │ │ └── nas_fcos_nashead_r50_caffe_fpn_gn-head_4x4_1x_coco.py │ │ ├── nas_fpn │ │ │ ├── README.md │ │ │ ├── metafile.yml │ │ │ ├── retinanet_r50_fpn_crop640_50e_coco.py │ │ │ └── retinanet_r50_nasfpn_crop640_50e_coco.py │ │ ├── openimages │ │ │ ├── README.md │ │ │ ├── faster_rcnn_r50_fpn_32x2_1x_openimages.py │ │ │ ├── faster_rcnn_r50_fpn_32x2_1x_openimages_challenge.py │ │ │ ├── faster_rcnn_r50_fpn_32x2_cas_1x_openimages.py │ │ │ ├── faster_rcnn_r50_fpn_32x2_cas_1x_openimages_challenge.py │ │ │ ├── metafile.yml │ │ │ ├── retinanet_r50_fpn_32x2_1x_openimages.py │ │ │ └── ssd300_32x8_36e_openimages.py │ │ ├── paa │ │ │ ├── README.md │ │ │ ├── metafile.yml │ │ │ ├── paa_r101_fpn_1x_coco.py │ │ │ ├── paa_r101_fpn_2x_coco.py │ │ │ ├── paa_r101_fpn_mstrain_3x_coco.py │ │ │ ├── paa_r50_fpn_1.5x_coco.py │ │ │ ├── paa_r50_fpn_1x_coco.py │ │ │ ├── paa_r50_fpn_2x_coco.py │ │ │ └── paa_r50_fpn_mstrain_3x_coco.py │ │ ├── pafpn │ │ │ ├── README.md │ │ │ ├── faster_rcnn_r50_pafpn_1x_coco.py │ │ │ └── metafile.yml │ │ ├── panoptic_fpn │ │ │ ├── README.md │ │ │ ├── metafile.yml │ │ │ ├── panoptic_fpn_r101_fpn_1x_coco.py │ │ │ ├── panoptic_fpn_r101_fpn_mstrain_3x_coco.py │ │ │ ├── panoptic_fpn_r50_fpn_1x_coco.py │ │ │ └── panoptic_fpn_r50_fpn_mstrain_3x_coco.py │ │ ├── pascal_voc │ │ │ ├── README.md │ │ │ ├── faster_rcnn_r50_caffe_c4_mstrain_18k_voc0712.py │ │ │ ├── faster_rcnn_r50_fpn_1x_voc0712.py │ │ │ ├── faster_rcnn_r50_fpn_1x_voc0712_cocofmt.py │ │ │ ├── retinanet_r50_fpn_1x_voc0712.py │ │ │ ├── ssd300_voc0712.py │ │ │ └── ssd512_voc0712.py │ │ ├── pisa │ │ │ ├── README.md │ │ │ ├── metafile.yml │ │ │ ├── pisa_faster_rcnn_r50_fpn_1x_coco.py │ │ │ ├── pisa_faster_rcnn_x101_32x4d_fpn_1x_coco.py │ │ │ ├── pisa_mask_rcnn_r50_fpn_1x_coco.py │ │ │ ├── pisa_mask_rcnn_x101_32x4d_fpn_1x_coco.py │ │ │ ├── pisa_retinanet_r50_fpn_1x_coco.py │ │ │ ├── pisa_retinanet_x101_32x4d_fpn_1x_coco.py │ │ │ ├── pisa_ssd300_coco.py │ │ │ └── pisa_ssd512_coco.py │ │ ├── point_rend │ │ │ ├── README.md │ │ │ ├── metafile.yml │ │ │ ├── point_rend_r50_caffe_fpn_mstrain_1x_coco.py │ │ │ └── point_rend_r50_caffe_fpn_mstrain_3x_coco.py │ │ ├── pvt │ │ │ ├── README.md │ │ │ ├── metafile.yml │ │ │ ├── retinanet_pvt-l_fpn_1x_coco.py │ │ │ ├── retinanet_pvt-m_fpn_1x_coco.py │ │ │ ├── retinanet_pvt-s_fpn_1x_coco.py │ │ │ ├── retinanet_pvt-t_fpn_1x_coco.py │ │ │ ├── retinanet_pvtv2-b0_fpn_1x_coco.py │ │ │ ├── retinanet_pvtv2-b1_fpn_1x_coco.py │ │ │ ├── retinanet_pvtv2-b2_fpn_1x_coco.py │ │ │ ├── retinanet_pvtv2-b3_fpn_1x_coco.py │ │ │ ├── retinanet_pvtv2-b4_fpn_1x_coco.py │ │ │ └── retinanet_pvtv2-b5_fpn_1x_coco.py │ │ ├── queryinst │ │ │ ├── README.md │ │ │ ├── metafile.yml │ │ │ ├── queryinst_r101_fpn_300_proposals_crop_mstrain_480-800_3x_coco.py │ │ │ ├── queryinst_r101_fpn_mstrain_480-800_3x_coco.py │ │ │ ├── queryinst_r50_fpn_1x_coco.py │ │ │ ├── queryinst_r50_fpn_300_proposals_crop_mstrain_480-800_3x_coco.py │ │ │ └── queryinst_r50_fpn_mstrain_480-800_3x_coco.py │ │ ├── regnet │ │ │ ├── README.md │ │ │ ├── cascade_mask_rcnn_regnetx-1.6GF_fpn_mstrain_3x_coco.py │ │ │ ├── cascade_mask_rcnn_regnetx-3.2GF_fpn_mstrain_3x_coco.py │ │ │ ├── cascade_mask_rcnn_regnetx-400MF_fpn_mstrain_3x_coco.py │ │ │ ├── cascade_mask_rcnn_regnetx-4GF_fpn_mstrain_3x_coco.py │ │ │ ├── cascade_mask_rcnn_regnetx-800MF_fpn_mstrain_3x_coco.py │ │ │ ├── faster_rcnn_regnetx-1.6GF_fpn_mstrain_3x_coco.py │ │ │ ├── faster_rcnn_regnetx-3.2GF_fpn_1x_coco.py │ │ │ ├── faster_rcnn_regnetx-3.2GF_fpn_2x_coco.py │ │ │ ├── faster_rcnn_regnetx-3.2GF_fpn_mstrain_3x_coco.py │ │ │ ├── faster_rcnn_regnetx-400MF_fpn_mstrain_3x_coco.py │ │ │ ├── faster_rcnn_regnetx-4GF_fpn_mstrain_3x_coco.py │ │ │ ├── faster_rcnn_regnetx-800MF_fpn_mstrain_3x_coco.py │ │ │ ├── mask_rcnn_regnetx-1.6GF_fpn_mstrain-poly_3x_coco.py │ │ │ ├── mask_rcnn_regnetx-12GF_fpn_1x_coco.py │ │ │ ├── mask_rcnn_regnetx-3.2GF_fpn_1x_coco.py │ │ │ ├── mask_rcnn_regnetx-3.2GF_fpn_mdconv_c3-c5_1x_coco.py │ │ │ ├── mask_rcnn_regnetx-3.2GF_fpn_mstrain_3x_coco.py │ │ │ ├── mask_rcnn_regnetx-400MF_fpn_mstrain-poly_3x_coco.py │ │ │ ├── mask_rcnn_regnetx-4GF_fpn_1x_coco.py │ │ │ ├── mask_rcnn_regnetx-4GF_fpn_mstrain-poly_3x_coco.py │ │ │ ├── mask_rcnn_regnetx-6.4GF_fpn_1x_coco.py │ │ │ ├── mask_rcnn_regnetx-800MF_fpn_mstrain-poly_3x_coco.py │ │ │ ├── mask_rcnn_regnetx-8GF_fpn_1x_coco.py │ │ │ ├── metafile.yml │ │ │ ├── retinanet_regnetx-1.6GF_fpn_1x_coco.py │ │ │ ├── retinanet_regnetx-3.2GF_fpn_1x_coco.py │ │ │ └── retinanet_regnetx-800MF_fpn_1x_coco.py │ │ ├── reppoints │ │ │ ├── README.md │ │ │ ├── bbox_r50_grid_center_fpn_gn-neck+head_1x_coco.py │ │ │ ├── bbox_r50_grid_fpn_gn-neck+head_1x_coco.py │ │ │ ├── metafile.yml │ │ │ ├── reppoints_minmax_r50_fpn_gn-neck+head_1x_coco.py │ │ │ ├── reppoints_moment_r101_fpn_dconv_c3-c5_gn-neck+head_2x_coco.py │ │ │ ├── reppoints_moment_r101_fpn_gn-neck+head_2x_coco.py │ │ │ ├── reppoints_moment_r50_fpn_1x_coco.py │ │ │ ├── reppoints_moment_r50_fpn_gn-neck+head_1x_coco.py │ │ │ ├── reppoints_moment_r50_fpn_gn-neck+head_2x_coco.py │ │ │ ├── reppoints_moment_x101_fpn_dconv_c3-c5_gn-neck+head_2x_coco.py │ │ │ └── reppoints_partial_minmax_r50_fpn_gn-neck+head_1x_coco.py │ │ ├── res2net │ │ │ ├── README.md │ │ │ ├── cascade_mask_rcnn_r2_101_fpn_20e_coco.py │ │ │ ├── cascade_rcnn_r2_101_fpn_20e_coco.py │ │ │ ├── faster_rcnn_r2_101_fpn_2x_coco.py │ │ │ ├── htc_r2_101_fpn_20e_coco.py │ │ │ ├── mask_rcnn_r2_101_fpn_2x_coco.py │ │ │ └── metafile.yml │ │ ├── resnest │ │ │ ├── README.md │ │ │ ├── cascade_mask_rcnn_s101_fpn_syncbn-backbone+head_mstrain_1x_coco.py │ │ │ ├── cascade_mask_rcnn_s50_fpn_syncbn-backbone+head_mstrain_1x_coco.py │ │ │ ├── cascade_rcnn_s101_fpn_syncbn-backbone+head_mstrain-range_1x_coco.py │ │ │ ├── cascade_rcnn_s50_fpn_syncbn-backbone+head_mstrain-range_1x_coco.py │ │ │ ├── faster_rcnn_s101_fpn_syncbn-backbone+head_mstrain-range_1x_coco.py │ │ │ ├── faster_rcnn_s50_fpn_syncbn-backbone+head_mstrain-range_1x_coco.py │ │ │ ├── mask_rcnn_s101_fpn_syncbn-backbone+head_mstrain_1x_coco.py │ │ │ ├── mask_rcnn_s50_fpn_syncbn-backbone+head_mstrain_1x_coco.py │ │ │ └── metafile.yml │ │ ├── resnet_strikes_back │ │ │ ├── README.md │ │ │ ├── cascade_mask_rcnn_r50_fpn_rsb-pretrain_1x_coco.py │ │ │ ├── faster_rcnn_r50_fpn_rsb-pretrain_1x_coco.py │ │ │ ├── mask_rcnn_r50_fpn_rsb-pretrain_1x_coco.py │ │ │ ├── metafile.yml │ │ │ └── retinanet_r50_fpn_rsb-pretrain_1x_coco.py │ │ ├── retinanet │ │ │ ├── README.md │ │ │ ├── metafile.yml │ │ │ ├── retinanet_r101_caffe_fpn_1x_coco.py │ │ │ ├── retinanet_r101_caffe_fpn_mstrain_3x_coco.py │ │ │ ├── retinanet_r101_fpn_1x_coco.py │ │ │ ├── retinanet_r101_fpn_2x_coco.py │ │ │ ├── retinanet_r101_fpn_mstrain_640-800_3x_coco.py │ │ │ ├── retinanet_r18_fpn_1x8_1x_coco.py │ │ │ ├── retinanet_r18_fpn_1x_coco.py │ │ │ ├── retinanet_r50_caffe_fpn_1x_coco.py │ │ │ ├── retinanet_r50_caffe_fpn_mstrain_1x_coco.py │ │ │ ├── retinanet_r50_caffe_fpn_mstrain_2x_coco.py │ │ │ ├── retinanet_r50_caffe_fpn_mstrain_3x_coco.py │ │ │ ├── retinanet_r50_fpn_1x_coco.py │ │ │ ├── retinanet_r50_fpn_2x_coco.py │ │ │ ├── retinanet_r50_fpn_90k_coco.py │ │ │ ├── retinanet_r50_fpn_fp16_1x_coco.py │ │ │ ├── retinanet_r50_fpn_mstrain_640-800_3x_coco.py │ │ │ ├── retinanet_x101_32x4d_fpn_1x_coco.py │ │ │ ├── retinanet_x101_32x4d_fpn_2x_coco.py │ │ │ ├── retinanet_x101_64x4d_fpn_1x_coco.py │ │ │ ├── retinanet_x101_64x4d_fpn_2x_coco.py │ │ │ └── retinanet_x101_64x4d_fpn_mstrain_640-800_3x_coco.py │ │ ├── rpn │ │ │ ├── README.md │ │ │ ├── rpn_r101_caffe_fpn_1x_coco.py │ │ │ ├── rpn_r101_fpn_1x_coco.py │ │ │ ├── rpn_r101_fpn_2x_coco.py │ │ │ ├── rpn_r50_caffe_c4_1x_coco.py │ │ │ ├── rpn_r50_caffe_fpn_1x_coco.py │ │ │ ├── rpn_r50_fpn_1x_coco.py │ │ │ ├── rpn_r50_fpn_2x_coco.py │ │ │ ├── rpn_x101_32x4d_fpn_1x_coco.py │ │ │ ├── rpn_x101_32x4d_fpn_2x_coco.py │ │ │ ├── rpn_x101_64x4d_fpn_1x_coco.py │ │ │ └── rpn_x101_64x4d_fpn_2x_coco.py │ │ ├── sabl │ │ │ ├── README.md │ │ │ ├── metafile.yml │ │ │ ├── sabl_cascade_rcnn_r101_fpn_1x_coco.py │ │ │ ├── sabl_cascade_rcnn_r50_fpn_1x_coco.py │ │ │ ├── sabl_faster_rcnn_r101_fpn_1x_coco.py │ │ │ ├── sabl_faster_rcnn_r50_fpn_1x_coco.py │ │ │ ├── sabl_retinanet_r101_fpn_1x_coco.py │ │ │ ├── sabl_retinanet_r101_fpn_gn_1x_coco.py │ │ │ ├── sabl_retinanet_r101_fpn_gn_2x_ms_480_960_coco.py │ │ │ ├── sabl_retinanet_r101_fpn_gn_2x_ms_640_800_coco.py │ │ │ ├── sabl_retinanet_r50_fpn_1x_coco.py │ │ │ └── sabl_retinanet_r50_fpn_gn_1x_coco.py │ │ ├── scnet │ │ │ ├── README.md │ │ │ ├── metafile.yml │ │ │ ├── scnet_r101_fpn_20e_coco.py │ │ │ ├── scnet_r50_fpn_1x_coco.py │ │ │ ├── scnet_r50_fpn_20e_coco.py │ │ │ ├── scnet_x101_64x4d_fpn_20e_coco.py │ │ │ └── scnet_x101_64x4d_fpn_8x1_20e_coco.py │ │ ├── scratch │ │ │ ├── README.md │ │ │ ├── faster_rcnn_r50_fpn_gn-all_scratch_6x_coco.py │ │ │ ├── mask_rcnn_r50_fpn_gn-all_scratch_6x_coco.py │ │ │ └── metafile.yml │ │ ├── seesaw_loss │ │ │ ├── README.md │ │ │ ├── cascade_mask_rcnn_r101_fpn_random_seesaw_loss_mstrain_2x_lvis_v1.py │ │ │ ├── cascade_mask_rcnn_r101_fpn_random_seesaw_loss_normed_mask_mstrain_2x_lvis_v1.py │ │ │ ├── cascade_mask_rcnn_r101_fpn_sample1e-3_seesaw_loss_mstrain_2x_lvis_v1.py │ │ │ ├── cascade_mask_rcnn_r101_fpn_sample1e-3_seesaw_loss_normed_mask_mstrain_2x_lvis_v1.py │ │ │ ├── mask_rcnn_r101_fpn_random_seesaw_loss_mstrain_2x_lvis_v1.py │ │ │ ├── mask_rcnn_r101_fpn_random_seesaw_loss_normed_mask_mstrain_2x_lvis_v1.py │ │ │ ├── mask_rcnn_r101_fpn_sample1e-3_seesaw_loss_mstrain_2x_lvis_v1.py │ │ │ ├── mask_rcnn_r101_fpn_sample1e-3_seesaw_loss_normed_mask_mstrain_2x_lvis_v1.py │ │ │ ├── mask_rcnn_r50_fpn_random_seesaw_loss_mstrain_2x_lvis_v1.py │ │ │ ├── mask_rcnn_r50_fpn_random_seesaw_loss_normed_mask_mstrain_2x_lvis_v1.py │ │ │ ├── mask_rcnn_r50_fpn_sample1e-3_seesaw_loss_mstrain_2x_lvis_v1.py │ │ │ ├── mask_rcnn_r50_fpn_sample1e-3_seesaw_loss_normed_mask_mstrain_2x_lvis_v1.py │ │ │ └── metafile.yml │ │ ├── selfsup_pretrain │ │ │ ├── README.md │ │ │ ├── mask_rcnn_r50_fpn_mocov2-pretrain_1x_coco.py │ │ │ ├── mask_rcnn_r50_fpn_mocov2-pretrain_ms-2x_coco.py │ │ │ ├── mask_rcnn_r50_fpn_swav-pretrain_1x_coco.py │ │ │ └── mask_rcnn_r50_fpn_swav-pretrain_ms-2x_coco.py │ │ ├── simple_copy_paste │ │ │ ├── README.md │ │ │ ├── mask_rcnn_r50_fpn_syncbn-all_rpn-2conv_ssj_32x2_270k_coco.py │ │ │ ├── mask_rcnn_r50_fpn_syncbn-all_rpn-2conv_ssj_32x2_90k_coco.py │ │ │ ├── mask_rcnn_r50_fpn_syncbn-all_rpn-2conv_ssj_scp_32x2_270k_coco.py │ │ │ ├── mask_rcnn_r50_fpn_syncbn-all_rpn-2conv_ssj_scp_32x2_90k_coco.py │ │ │ └── metafile.yml │ │ ├── solo │ │ │ ├── README.md │ │ │ ├── decoupled_solo_light_r50_fpn_3x_coco.py │ │ │ ├── decoupled_solo_r50_fpn_1x_coco.py │ │ │ ├── decoupled_solo_r50_fpn_3x_coco.py │ │ │ ├── metafile.yml │ │ │ ├── solo_r50_fpn_1x_coco.py │ │ │ └── solo_r50_fpn_3x_coco.py │ │ ├── solov2 │ │ │ ├── README.md │ │ │ ├── metafile.yml │ │ │ ├── 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 │ │ ├── sparse_rcnn │ │ │ ├── README.md │ │ │ ├── metafile.yml │ │ │ ├── sparse_rcnn_r101_fpn_300_proposals_crop_mstrain_480-800_3x_coco.py │ │ │ ├── sparse_rcnn_r101_fpn_mstrain_480-800_3x_coco.py │ │ │ ├── sparse_rcnn_r50_fpn_1x_coco.py │ │ │ ├── sparse_rcnn_r50_fpn_300_proposals_crop_mstrain_480-800_3x_coco.py │ │ │ └── sparse_rcnn_r50_fpn_mstrain_480-800_3x_coco.py │ │ ├── ssd │ │ │ ├── README.md │ │ │ ├── metafile.yml │ │ │ ├── ssd300_coco.py │ │ │ ├── ssd512_coco.py │ │ │ └── ssdlite_mobilenetv2_scratch_600e_coco.py │ │ ├── strong_baselines │ │ │ ├── README.md │ │ │ ├── mask_rcnn_r50_caffe_fpn_syncbn-all_rpn-2conv_lsj_100e_coco.py │ │ │ ├── mask_rcnn_r50_caffe_fpn_syncbn-all_rpn-2conv_lsj_100e_fp16_coco.py │ │ │ ├── mask_rcnn_r50_caffe_fpn_syncbn-all_rpn-2conv_lsj_400e_coco.py │ │ │ ├── mask_rcnn_r50_fpn_syncbn-all_rpn-2conv_lsj_100e_coco.py │ │ │ ├── mask_rcnn_r50_fpn_syncbn-all_rpn-2conv_lsj_100e_fp16_coco.py │ │ │ └── mask_rcnn_r50_fpn_syncbn-all_rpn-2conv_lsj_50e_coco.py │ │ ├── swin │ │ │ ├── README.md │ │ │ ├── mask_rcnn_swin-s-p4-w7_fpn_fp16_ms-crop-3x_coco.py │ │ │ ├── mask_rcnn_swin-t-p4-w7_fpn_1x_coco.py │ │ │ ├── mask_rcnn_swin-t-p4-w7_fpn_fp16_ms-crop-3x_coco.py │ │ │ ├── mask_rcnn_swin-t-p4-w7_fpn_ms-crop-3x_coco.py │ │ │ ├── metafile.yml │ │ │ └── retinanet_swin-t-p4-w7_fpn_1x_coco.py │ │ ├── timm_example │ │ │ ├── README.md │ │ │ ├── retinanet_timm_efficientnet_b1_fpn_1x_coco.py │ │ │ └── retinanet_timm_tv_resnet50_fpn_1x_coco.py │ │ ├── tood │ │ │ ├── README.md │ │ │ ├── metafile.yml │ │ │ ├── tood_r101_fpn_dconv_c3-c5_mstrain_2x_coco.py │ │ │ ├── tood_r101_fpn_mstrain_2x_coco.py │ │ │ ├── tood_r50_fpn_1x_coco.py │ │ │ ├── tood_r50_fpn_anchor_based_1x_coco.py │ │ │ ├── tood_r50_fpn_mstrain_2x_coco.py │ │ │ ├── tood_x101_64x4d_fpn_dconv_c4-c5_mstrain_2x_coco.py │ │ │ └── tood_x101_64x4d_fpn_mstrain_2x_coco.py │ │ ├── tridentnet │ │ │ ├── README.md │ │ │ ├── metafile.yml │ │ │ ├── tridentnet_r50_caffe_1x_coco.py │ │ │ ├── tridentnet_r50_caffe_mstrain_1x_coco.py │ │ │ └── tridentnet_r50_caffe_mstrain_3x_coco.py │ │ ├── vfnet │ │ │ ├── README.md │ │ │ ├── metafile.yml │ │ │ ├── vfnet_r101_fpn_1x_coco.py │ │ │ ├── vfnet_r101_fpn_2x_coco.py │ │ │ ├── vfnet_r101_fpn_mdconv_c3-c5_mstrain_2x_coco.py │ │ │ ├── vfnet_r101_fpn_mstrain_2x_coco.py │ │ │ ├── vfnet_r2_101_fpn_mdconv_c3-c5_mstrain_2x_coco.py │ │ │ ├── vfnet_r2_101_fpn_mstrain_2x_coco.py │ │ │ ├── vfnet_r50_fpn_1x_coco.py │ │ │ ├── vfnet_r50_fpn_mdconv_c3-c5_mstrain_2x_coco.py │ │ │ ├── vfnet_r50_fpn_mstrain_2x_coco.py │ │ │ ├── vfnet_x101_32x4d_fpn_mdconv_c3-c5_mstrain_2x_coco.py │ │ │ ├── vfnet_x101_32x4d_fpn_mstrain_2x_coco.py │ │ │ ├── vfnet_x101_64x4d_fpn_mdconv_c3-c5_mstrain_2x_coco.py │ │ │ └── vfnet_x101_64x4d_fpn_mstrain_2x_coco.py │ │ ├── wider_face │ │ │ ├── README.md │ │ │ └── ssd300_wider_face.py │ │ ├── yolact │ │ │ ├── README.md │ │ │ ├── metafile.yml │ │ │ ├── yolact_r101_1x8_coco.py │ │ │ ├── yolact_r50_1x8_coco.py │ │ │ └── yolact_r50_8x8_coco.py │ │ ├── yolo │ │ │ ├── README.md │ │ │ ├── metafile.yml │ │ │ ├── yolov3_d53_320_273e_coco.py │ │ │ ├── yolov3_d53_fp16_mstrain-608_273e_coco.py │ │ │ ├── yolov3_d53_mstrain-416_273e_coco.py │ │ │ ├── yolov3_d53_mstrain-608_273e_coco.py │ │ │ ├── yolov3_mobilenetv2_320_300e_coco.py │ │ │ └── yolov3_mobilenetv2_mstrain-416_300e_coco.py │ │ ├── yolof │ │ │ ├── README.md │ │ │ ├── metafile.yml │ │ │ ├── yolof_r50_c5_8x8_1x_coco.py │ │ │ └── yolof_r50_c5_8x8_iter-1x_coco.py │ │ └── yolox │ │ │ ├── README.md │ │ │ ├── metafile.yml │ │ │ ├── yolox_l_8x8_300e_coco.py │ │ │ ├── yolox_m_8x8_300e_coco.py │ │ │ ├── yolox_nano_8x8_300e_coco.py │ │ │ ├── yolox_s_8x8_300e_coco.py │ │ │ ├── yolox_tiny_8x8_300e_coco.py │ │ │ └── yolox_x_8x8_300e_coco.py │ ├── demo │ │ ├── create_result_gif.py │ │ ├── demo.mp4 │ │ ├── image_demo.py │ │ ├── video_demo.py │ │ ├── video_gpuaccel_demo.py │ │ └── webcam_demo.py │ ├── docker │ │ ├── Dockerfile │ │ └── serve │ │ │ ├── Dockerfile │ │ │ ├── config.properties │ │ │ └── entrypoint.sh │ ├── docs │ │ ├── en │ │ │ ├── 1_exist_data_model.md │ │ │ ├── 2_new_data_model.md │ │ │ ├── 3_exist_data_new_model.md │ │ │ ├── Makefile │ │ │ ├── _static │ │ │ │ └── css │ │ │ │ │ └── readthedocs.css │ │ │ ├── api.rst │ │ │ ├── changelog.md │ │ │ ├── compatibility.md │ │ │ ├── conf.py │ │ │ ├── conventions.md │ │ │ ├── faq.md │ │ │ ├── get_started.md │ │ │ ├── index.rst │ │ │ ├── make.bat │ │ │ ├── model_zoo.md │ │ │ ├── projects.md │ │ │ ├── robustness_benchmarking.md │ │ │ ├── stat.py │ │ │ ├── switch_language.md │ │ │ ├── tutorials │ │ │ │ ├── config.md │ │ │ │ ├── customize_dataset.md │ │ │ │ ├── customize_losses.md │ │ │ │ ├── customize_models.md │ │ │ │ ├── customize_runtime.md │ │ │ │ ├── data_pipeline.md │ │ │ │ ├── finetune.md │ │ │ │ ├── how_to.md │ │ │ │ ├── index.rst │ │ │ │ ├── init_cfg.md │ │ │ │ ├── onnx2tensorrt.md │ │ │ │ ├── pytorch2onnx.md │ │ │ │ ├── test_results_submission.md │ │ │ │ └── useful_hooks.md │ │ │ └── useful_tools.md │ │ └── zh_cn │ │ │ ├── 1_exist_data_model.md │ │ │ ├── 2_new_data_model.md │ │ │ ├── 3_exist_data_new_model.md │ │ │ ├── Makefile │ │ │ ├── _static │ │ │ └── css │ │ │ │ └── readthedocs.css │ │ │ ├── api.rst │ │ │ ├── article.md │ │ │ ├── compatibility.md │ │ │ ├── conf.py │ │ │ ├── conventions.md │ │ │ ├── faq.md │ │ │ ├── get_started.md │ │ │ ├── index.rst │ │ │ ├── make.bat │ │ │ ├── model_zoo.md │ │ │ ├── projects.md │ │ │ ├── robustness_benchmarking.md │ │ │ ├── stat.py │ │ │ ├── switch_language.md │ │ │ ├── tutorials │ │ │ ├── config.md │ │ │ ├── customize_dataset.md │ │ │ ├── customize_losses.md │ │ │ ├── customize_models.md │ │ │ ├── customize_runtime.md │ │ │ ├── data_pipeline.md │ │ │ ├── finetune.md │ │ │ ├── how_to.md │ │ │ ├── index.rst │ │ │ ├── init_cfg.md │ │ │ ├── onnx2tensorrt.md │ │ │ └── pytorch2onnx.md │ │ │ └── useful_tools.md │ ├── mmdet │ │ ├── .mim │ │ │ ├── configs │ │ │ ├── demo │ │ │ ├── model-index.yml │ │ │ └── tools │ │ ├── __init__.py │ │ ├── apis │ │ │ ├── __init__.py │ │ │ ├── inference.py │ │ │ ├── test.py │ │ │ └── train.py │ │ ├── core │ │ │ ├── __init__.py │ │ │ ├── anchor │ │ │ │ ├── __init__.py │ │ │ │ ├── anchor_generator.py │ │ │ │ ├── builder.py │ │ │ │ ├── point_generator.py │ │ │ │ └── utils.py │ │ │ ├── bbox │ │ │ │ ├── __init__.py │ │ │ │ ├── assigners │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── approx_max_iou_assigner.py │ │ │ │ │ ├── assign_result.py │ │ │ │ │ ├── atss_assigner.py │ │ │ │ │ ├── base_assigner.py │ │ │ │ │ ├── center_region_assigner.py │ │ │ │ │ ├── grid_assigner.py │ │ │ │ │ ├── hungarian_assigner.py │ │ │ │ │ ├── mask_hungarian_assigner.py │ │ │ │ │ ├── max_iou_assigner.py │ │ │ │ │ ├── point_assigner.py │ │ │ │ │ ├── region_assigner.py │ │ │ │ │ ├── sim_ota_assigner.py │ │ │ │ │ ├── task_aligned_assigner.py │ │ │ │ │ └── uniform_assigner.py │ │ │ │ ├── builder.py │ │ │ │ ├── coder │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── base_bbox_coder.py │ │ │ │ │ ├── bucketing_bbox_coder.py │ │ │ │ │ ├── delta_xywh_bbox_coder.py │ │ │ │ │ ├── distance_point_bbox_coder.py │ │ │ │ │ ├── legacy_delta_xywh_bbox_coder.py │ │ │ │ │ ├── pseudo_bbox_coder.py │ │ │ │ │ ├── tblr_bbox_coder.py │ │ │ │ │ └── yolo_bbox_coder.py │ │ │ │ ├── demodata.py │ │ │ │ ├── iou_calculators │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── builder.py │ │ │ │ │ └── iou2d_calculator.py │ │ │ │ ├── match_costs │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── builder.py │ │ │ │ │ └── match_cost.py │ │ │ │ ├── samplers │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── base_sampler.py │ │ │ │ │ ├── combined_sampler.py │ │ │ │ │ ├── instance_balanced_pos_sampler.py │ │ │ │ │ ├── iou_balanced_neg_sampler.py │ │ │ │ │ ├── mask_pseudo_sampler.py │ │ │ │ │ ├── mask_sampling_result.py │ │ │ │ │ ├── ohem_sampler.py │ │ │ │ │ ├── pseudo_sampler.py │ │ │ │ │ ├── random_sampler.py │ │ │ │ │ ├── sampling_result.py │ │ │ │ │ └── score_hlr_sampler.py │ │ │ │ └── transforms.py │ │ │ ├── data_structures │ │ │ │ ├── __init__.py │ │ │ │ ├── general_data.py │ │ │ │ └── instance_data.py │ │ │ ├── evaluation │ │ │ │ ├── __init__.py │ │ │ │ ├── bbox_overlaps.py │ │ │ │ ├── class_names.py │ │ │ │ ├── eval_hooks.py │ │ │ │ ├── mean_ap.py │ │ │ │ ├── panoptic_utils.py │ │ │ │ └── recall.py │ │ │ ├── export │ │ │ │ ├── __init__.py │ │ │ │ ├── model_wrappers.py │ │ │ │ ├── onnx_helper.py │ │ │ │ └── pytorch2onnx.py │ │ │ ├── hook │ │ │ │ ├── __init__.py │ │ │ │ ├── checkloss_hook.py │ │ │ │ ├── ema.py │ │ │ │ ├── memory_profiler_hook.py │ │ │ │ ├── set_epoch_info_hook.py │ │ │ │ ├── sync_norm_hook.py │ │ │ │ ├── sync_random_size_hook.py │ │ │ │ ├── wandblogger_hook.py │ │ │ │ ├── yolox_lrupdater_hook.py │ │ │ │ └── yolox_mode_switch_hook.py │ │ │ ├── mask │ │ │ │ ├── __init__.py │ │ │ │ ├── mask_target.py │ │ │ │ ├── structures.py │ │ │ │ └── utils.py │ │ │ ├── optimizers │ │ │ │ ├── __init__.py │ │ │ │ ├── builder.py │ │ │ │ └── layer_decay_optimizer_constructor.py │ │ │ ├── post_processing │ │ │ │ ├── __init__.py │ │ │ │ ├── bbox_nms.py │ │ │ │ ├── matrix_nms.py │ │ │ │ └── merge_augs.py │ │ │ ├── utils │ │ │ │ ├── __init__.py │ │ │ │ ├── dist_utils.py │ │ │ │ └── misc.py │ │ │ └── visualization │ │ │ │ ├── __init__.py │ │ │ │ ├── image.py │ │ │ │ └── palette.py │ │ ├── datasets │ │ │ ├── __init__.py │ │ │ ├── api_wrappers │ │ │ │ ├── __init__.py │ │ │ │ ├── coco_api.py │ │ │ │ └── panoptic_evaluation.py │ │ │ ├── builder.py │ │ │ ├── cityscapes.py │ │ │ ├── coco.py │ │ │ ├── coco_panoptic.py │ │ │ ├── custom.py │ │ │ ├── dataset_wrappers.py │ │ │ ├── deepfashion.py │ │ │ ├── lvis.py │ │ │ ├── openimages.py │ │ │ ├── pipelines │ │ │ │ ├── __init__.py │ │ │ │ ├── auto_augment.py │ │ │ │ ├── compose.py │ │ │ │ ├── formating.py │ │ │ │ ├── formatting.py │ │ │ │ ├── instaboost.py │ │ │ │ ├── loading.py │ │ │ │ ├── test_time_aug.py │ │ │ │ └── transforms.py │ │ │ ├── samplers │ │ │ │ ├── __init__.py │ │ │ │ ├── class_aware_sampler.py │ │ │ │ ├── distributed_sampler.py │ │ │ │ ├── group_sampler.py │ │ │ │ └── infinite_sampler.py │ │ │ ├── utils.py │ │ │ ├── voc.py │ │ │ ├── wider_face.py │ │ │ └── xml_style.py │ │ ├── models │ │ │ ├── __init__.py │ │ │ ├── backbones │ │ │ │ ├── __init__.py │ │ │ │ ├── csp_darknet.py │ │ │ │ ├── darknet.py │ │ │ │ ├── detectors_resnet.py │ │ │ │ ├── detectors_resnext.py │ │ │ │ ├── efficientnet.py │ │ │ │ ├── hourglass.py │ │ │ │ ├── hrnet.py │ │ │ │ ├── mobilenet_v2.py │ │ │ │ ├── pvt.py │ │ │ │ ├── regnet.py │ │ │ │ ├── res2net.py │ │ │ │ ├── resnest.py │ │ │ │ ├── resnet.py │ │ │ │ ├── resnext.py │ │ │ │ ├── ssd_vgg.py │ │ │ │ ├── swin.py │ │ │ │ └── trident_resnet.py │ │ │ ├── builder.py │ │ │ ├── dense_heads │ │ │ │ ├── __init__.py │ │ │ │ ├── anchor_free_head.py │ │ │ │ ├── anchor_head.py │ │ │ │ ├── atss_head.py │ │ │ │ ├── autoassign_head.py │ │ │ │ ├── base_dense_head.py │ │ │ │ ├── base_mask_head.py │ │ │ │ ├── cascade_rpn_head.py │ │ │ │ ├── centernet_head.py │ │ │ │ ├── centripetal_head.py │ │ │ │ ├── corner_head.py │ │ │ │ ├── ddod_head.py │ │ │ │ ├── deformable_detr_head.py │ │ │ │ ├── dense_test_mixins.py │ │ │ │ ├── detr_head.py │ │ │ │ ├── embedding_rpn_head.py │ │ │ │ ├── fcos_head.py │ │ │ │ ├── fovea_head.py │ │ │ │ ├── free_anchor_retina_head.py │ │ │ │ ├── fsaf_head.py │ │ │ │ ├── ga_retina_head.py │ │ │ │ ├── ga_rpn_head.py │ │ │ │ ├── gfl_head.py │ │ │ │ ├── guided_anchor_head.py │ │ │ │ ├── lad_head.py │ │ │ │ ├── ld_head.py │ │ │ │ ├── mask2former_head.py │ │ │ │ ├── maskformer_head.py │ │ │ │ ├── nasfcos_head.py │ │ │ │ ├── paa_head.py │ │ │ │ ├── pisa_retinanet_head.py │ │ │ │ ├── pisa_ssd_head.py │ │ │ │ ├── reppoints_head.py │ │ │ │ ├── retina_head.py │ │ │ │ ├── retina_sepbn_head.py │ │ │ │ ├── rpn_head.py │ │ │ │ ├── sabl_retina_head.py │ │ │ │ ├── solo_head.py │ │ │ │ ├── solov2_head.py │ │ │ │ ├── ssd_head.py │ │ │ │ ├── tood_head.py │ │ │ │ ├── vfnet_head.py │ │ │ │ ├── yolact_head.py │ │ │ │ ├── yolo_head.py │ │ │ │ ├── yolof_head.py │ │ │ │ └── yolox_head.py │ │ │ ├── detectors │ │ │ │ ├── __init__.py │ │ │ │ ├── atss.py │ │ │ │ ├── autoassign.py │ │ │ │ ├── base.py │ │ │ │ ├── cascade_rcnn.py │ │ │ │ ├── centernet.py │ │ │ │ ├── cornernet.py │ │ │ │ ├── ddod.py │ │ │ │ ├── deformable_detr.py │ │ │ │ ├── detr.py │ │ │ │ ├── fast_rcnn.py │ │ │ │ ├── faster_rcnn.py │ │ │ │ ├── fcos.py │ │ │ │ ├── fovea.py │ │ │ │ ├── fsaf.py │ │ │ │ ├── gfl.py │ │ │ │ ├── grid_rcnn.py │ │ │ │ ├── htc.py │ │ │ │ ├── kd_one_stage.py │ │ │ │ ├── lad.py │ │ │ │ ├── mask2former.py │ │ │ │ ├── mask_rcnn.py │ │ │ │ ├── mask_scoring_rcnn.py │ │ │ │ ├── maskformer.py │ │ │ │ ├── nasfcos.py │ │ │ │ ├── paa.py │ │ │ │ ├── panoptic_fpn.py │ │ │ │ ├── panoptic_two_stage_segmentor.py │ │ │ │ ├── point_rend.py │ │ │ │ ├── queryinst.py │ │ │ │ ├── reppoints_detector.py │ │ │ │ ├── retinanet.py │ │ │ │ ├── rpn.py │ │ │ │ ├── scnet.py │ │ │ │ ├── single_stage.py │ │ │ │ ├── single_stage_instance_seg.py │ │ │ │ ├── solo.py │ │ │ │ ├── solov2.py │ │ │ │ ├── sparse_rcnn.py │ │ │ │ ├── tood.py │ │ │ │ ├── trident_faster_rcnn.py │ │ │ │ ├── two_stage.py │ │ │ │ ├── vfnet.py │ │ │ │ ├── yolact.py │ │ │ │ ├── yolo.py │ │ │ │ ├── yolof.py │ │ │ │ └── yolox.py │ │ │ ├── losses │ │ │ │ ├── __init__.py │ │ │ │ ├── accuracy.py │ │ │ │ ├── ae_loss.py │ │ │ │ ├── balanced_l1_loss.py │ │ │ │ ├── cross_entropy_loss.py │ │ │ │ ├── dice_loss.py │ │ │ │ ├── focal_loss.py │ │ │ │ ├── gaussian_focal_loss.py │ │ │ │ ├── gfocal_loss.py │ │ │ │ ├── ghm_loss.py │ │ │ │ ├── iou_loss.py │ │ │ │ ├── kd_loss.py │ │ │ │ ├── mse_loss.py │ │ │ │ ├── pisa_loss.py │ │ │ │ ├── seesaw_loss.py │ │ │ │ ├── smooth_l1_loss.py │ │ │ │ ├── utils.py │ │ │ │ └── varifocal_loss.py │ │ │ ├── necks │ │ │ │ ├── __init__.py │ │ │ │ ├── bfp.py │ │ │ │ ├── channel_mapper.py │ │ │ │ ├── ct_resnet_neck.py │ │ │ │ ├── dilated_encoder.py │ │ │ │ ├── dyhead.py │ │ │ │ ├── fpg.py │ │ │ │ ├── fpn.py │ │ │ │ ├── fpn_carafe.py │ │ │ │ ├── hrfpn.py │ │ │ │ ├── nas_fpn.py │ │ │ │ ├── nasfcos_fpn.py │ │ │ │ ├── pafpn.py │ │ │ │ ├── rfp.py │ │ │ │ ├── ssd_neck.py │ │ │ │ ├── yolo_neck.py │ │ │ │ └── yolox_pafpn.py │ │ │ ├── plugins │ │ │ │ ├── __init__.py │ │ │ │ ├── dropblock.py │ │ │ │ ├── msdeformattn_pixel_decoder.py │ │ │ │ └── pixel_decoder.py │ │ │ ├── roi_heads │ │ │ │ ├── __init__.py │ │ │ │ ├── base_roi_head.py │ │ │ │ ├── bbox_heads │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── bbox_head.py │ │ │ │ │ ├── convfc_bbox_head.py │ │ │ │ │ ├── dii_head.py │ │ │ │ │ ├── double_bbox_head.py │ │ │ │ │ ├── sabl_head.py │ │ │ │ │ └── scnet_bbox_head.py │ │ │ │ ├── cascade_roi_head.py │ │ │ │ ├── double_roi_head.py │ │ │ │ ├── dynamic_roi_head.py │ │ │ │ ├── grid_roi_head.py │ │ │ │ ├── htc_roi_head.py │ │ │ │ ├── mask_heads │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── coarse_mask_head.py │ │ │ │ │ ├── dynamic_mask_head.py │ │ │ │ │ ├── fcn_mask_head.py │ │ │ │ │ ├── feature_relay_head.py │ │ │ │ │ ├── fused_semantic_head.py │ │ │ │ │ ├── global_context_head.py │ │ │ │ │ ├── grid_head.py │ │ │ │ │ ├── htc_mask_head.py │ │ │ │ │ ├── mask_point_head.py │ │ │ │ │ ├── maskiou_head.py │ │ │ │ │ ├── scnet_mask_head.py │ │ │ │ │ └── scnet_semantic_head.py │ │ │ │ ├── mask_scoring_roi_head.py │ │ │ │ ├── pisa_roi_head.py │ │ │ │ ├── point_rend_roi_head.py │ │ │ │ ├── roi_extractors │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── base_roi_extractor.py │ │ │ │ │ ├── generic_roi_extractor.py │ │ │ │ │ └── single_level_roi_extractor.py │ │ │ │ ├── scnet_roi_head.py │ │ │ │ ├── shared_heads │ │ │ │ │ ├── __init__.py │ │ │ │ │ └── res_layer.py │ │ │ │ ├── sparse_roi_head.py │ │ │ │ ├── standard_roi_head.py │ │ │ │ ├── test_mixins.py │ │ │ │ └── trident_roi_head.py │ │ │ ├── seg_heads │ │ │ │ ├── __init__.py │ │ │ │ ├── base_semantic_head.py │ │ │ │ ├── panoptic_fpn_head.py │ │ │ │ └── panoptic_fusion_heads │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── base_panoptic_fusion_head.py │ │ │ │ │ ├── heuristic_fusion_head.py │ │ │ │ │ └── maskformer_fusion_head.py │ │ │ └── utils │ │ │ │ ├── __init__.py │ │ │ │ ├── brick_wrappers.py │ │ │ │ ├── builder.py │ │ │ │ ├── ckpt_convert.py │ │ │ │ ├── conv_upsample.py │ │ │ │ ├── csp_layer.py │ │ │ │ ├── gaussian_target.py │ │ │ │ ├── inverted_residual.py │ │ │ │ ├── make_divisible.py │ │ │ │ ├── misc.py │ │ │ │ ├── normed_predictor.py │ │ │ │ ├── panoptic_gt_processing.py │ │ │ │ ├── point_sample.py │ │ │ │ ├── positional_encoding.py │ │ │ │ ├── res_layer.py │ │ │ │ ├── se_layer.py │ │ │ │ └── transformer.py │ │ ├── utils │ │ │ ├── __init__.py │ │ │ ├── collect_env.py │ │ │ ├── compat_config.py │ │ │ ├── contextmanagers.py │ │ │ ├── logger.py │ │ │ ├── memory.py │ │ │ ├── misc.py │ │ │ ├── profiling.py │ │ │ ├── replace_cfg_vals.py │ │ │ ├── setup_env.py │ │ │ ├── split_batch.py │ │ │ ├── util_distribution.py │ │ │ ├── util_mixins.py │ │ │ └── util_random.py │ │ └── version.py │ ├── model-index.yml │ ├── pytest.ini │ ├── requirements.txt │ ├── requirements │ │ ├── albu.txt │ │ ├── build.txt │ │ ├── docs.txt │ │ ├── mminstall.txt │ │ ├── optional.txt │ │ ├── readthedocs.txt │ │ ├── runtime.txt │ │ └── tests.txt │ ├── setup.cfg │ ├── setup.py │ ├── tests │ │ ├── test_data │ │ │ ├── test_datasets │ │ │ │ ├── test_coco_dataset.py │ │ │ │ ├── test_common.py │ │ │ │ ├── test_custom_dataset.py │ │ │ │ ├── test_dataset_wrapper.py │ │ │ │ ├── test_openimages_dataset.py │ │ │ │ ├── test_panoptic_dataset.py │ │ │ │ └── test_xml_dataset.py │ │ │ ├── test_pipelines │ │ │ │ ├── test_formatting.py │ │ │ │ ├── test_loading.py │ │ │ │ ├── test_sampler.py │ │ │ │ └── test_transform │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── test_img_augment.py │ │ │ │ │ ├── test_models_aug_test.py │ │ │ │ │ ├── test_rotate.py │ │ │ │ │ ├── test_shear.py │ │ │ │ │ ├── test_transform.py │ │ │ │ │ ├── test_translate.py │ │ │ │ │ └── utils.py │ │ │ └── test_utils.py │ │ ├── test_downstream │ │ │ └── test_mmtrack.py │ │ ├── test_metrics │ │ │ ├── test_box_overlap.py │ │ │ ├── test_losses.py │ │ │ ├── test_mean_ap.py │ │ │ └── test_recall.py │ │ ├── test_models │ │ │ ├── test_backbones │ │ │ │ ├── __init__.py │ │ │ │ ├── test_csp_darknet.py │ │ │ │ ├── test_detectors_resnet.py │ │ │ │ ├── test_efficientnet.py │ │ │ │ ├── test_hourglass.py │ │ │ │ ├── test_hrnet.py │ │ │ │ ├── test_mobilenet_v2.py │ │ │ │ ├── test_pvt.py │ │ │ │ ├── test_regnet.py │ │ │ │ ├── test_renext.py │ │ │ │ ├── test_res2net.py │ │ │ │ ├── test_resnest.py │ │ │ │ ├── test_resnet.py │ │ │ │ ├── test_swin.py │ │ │ │ ├── test_trident_resnet.py │ │ │ │ └── utils.py │ │ │ ├── test_dense_heads │ │ │ │ ├── test_anchor_head.py │ │ │ │ ├── test_atss_head.py │ │ │ │ ├── test_autoassign_head.py │ │ │ │ ├── test_centernet_head.py │ │ │ │ ├── test_corner_head.py │ │ │ │ ├── test_ddod_head.py │ │ │ │ ├── test_dense_heads_attr.py │ │ │ │ ├── test_detr_head.py │ │ │ │ ├── test_fcos_head.py │ │ │ │ ├── test_fsaf_head.py │ │ │ │ ├── test_ga_anchor_head.py │ │ │ │ ├── test_gfl_head.py │ │ │ │ ├── test_lad_head.py │ │ │ │ ├── test_ld_head.py │ │ │ │ ├── test_mask2former_head.py │ │ │ │ ├── test_maskformer_head.py │ │ │ │ ├── test_paa_head.py │ │ │ │ ├── test_pisa_head.py │ │ │ │ ├── test_sabl_retina_head.py │ │ │ │ ├── test_solo_head.py │ │ │ │ ├── test_tood_head.py │ │ │ │ ├── test_vfnet_head.py │ │ │ │ ├── test_yolact_head.py │ │ │ │ ├── test_yolof_head.py │ │ │ │ └── test_yolox_head.py │ │ │ ├── test_forward.py │ │ │ ├── test_loss.py │ │ │ ├── test_loss_compatibility.py │ │ │ ├── test_necks.py │ │ │ ├── test_plugins.py │ │ │ ├── test_roi_heads │ │ │ │ ├── __init__.py │ │ │ │ ├── test_bbox_head.py │ │ │ │ ├── test_mask_head.py │ │ │ │ ├── test_roi_extractor.py │ │ │ │ ├── test_sabl_bbox_head.py │ │ │ │ └── utils.py │ │ │ ├── test_seg_heads │ │ │ │ └── test_maskformer_fusion_head.py │ │ │ └── test_utils │ │ │ │ ├── test_brick_wrappers.py │ │ │ │ ├── test_conv_upsample.py │ │ │ │ ├── test_inverted_residual.py │ │ │ │ ├── test_model_misc.py │ │ │ │ ├── test_position_encoding.py │ │ │ │ ├── test_se_layer.py │ │ │ │ └── test_transformer.py │ │ ├── test_onnx │ │ │ ├── __init__.py │ │ │ ├── test_head.py │ │ │ ├── test_neck.py │ │ │ └── utils.py │ │ ├── test_runtime │ │ │ ├── async_benchmark.py │ │ │ ├── test_apis.py │ │ │ ├── test_async.py │ │ │ ├── test_config.py │ │ │ ├── test_eval_hook.py │ │ │ └── test_fp16.py │ │ └── test_utils │ │ │ ├── test_anchor.py │ │ │ ├── test_assigner.py │ │ │ ├── test_coder.py │ │ │ ├── test_compat_config.py │ │ │ ├── test_general_data.py │ │ │ ├── test_hook.py │ │ │ ├── test_layer_decay_optimizer_constructor.py │ │ │ ├── test_logger.py │ │ │ ├── test_masks.py │ │ │ ├── test_memory.py │ │ │ ├── test_misc.py │ │ │ ├── test_nms.py │ │ │ ├── test_replace_cfg_vals.py │ │ │ ├── test_setup_env.py │ │ │ ├── test_split_batch.py │ │ │ ├── test_version.py │ │ │ └── test_visualization.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 ├── mmdet3d │ ├── .dev_scripts │ │ ├── gather_models.py │ │ ├── gen_benchmark_script.py │ │ ├── linter.sh │ │ ├── test_benchmark.sh │ │ └── train_benchmark.sh │ ├── .github │ │ ├── CODE_OF_CONDUCT.md │ │ ├── CONTRIBUTING.md │ │ ├── ISSUE_TEMPLATE │ │ │ ├── config.yml │ │ │ ├── error-report.md │ │ │ ├── feature_request.md │ │ │ ├── general_questions.md │ │ │ └── reimplementation_questions.md │ │ ├── pull_request_template.md │ │ └── workflows │ │ │ ├── build.yml │ │ │ ├── deploy.yml │ │ │ └── lint.yml │ ├── .gitignore │ ├── .pre-commit-config.yaml │ ├── .readthedocs.yml │ ├── CITATION.cff │ ├── LICENSE │ ├── MANIFEST.in │ ├── README.md │ ├── README_zh-CN.md │ ├── configs │ │ ├── 3dssd │ │ │ ├── 3dssd_4x4_kitti-3d-car.py │ │ │ ├── README.md │ │ │ └── metafile.yml │ │ ├── _base_ │ │ │ ├── datasets │ │ │ │ ├── coco_instance.py │ │ │ │ ├── kitti-3d-3class.py │ │ │ │ ├── kitti-3d-car.py │ │ │ │ ├── kitti-mono3d.py │ │ │ │ ├── lyft-3d.py │ │ │ │ ├── nuim_instance.py │ │ │ │ ├── nus-3d.py │ │ │ │ ├── nus-mono3d.py │ │ │ │ ├── range100_lyft-3d.py │ │ │ │ ├── s3dis-3d-5class.py │ │ │ │ ├── s3dis_seg-3d-13class.py │ │ │ │ ├── scannet-3d-18class.py │ │ │ │ ├── scannet_seg-3d-20class.py │ │ │ │ ├── sunrgbd-3d-10class.py │ │ │ │ ├── waymoD5-3d-3class.py │ │ │ │ └── waymoD5-3d-car.py │ │ │ ├── default_runtime.py │ │ │ ├── models │ │ │ │ ├── 3dssd.py │ │ │ │ ├── cascade_mask_rcnn_r50_fpn.py │ │ │ │ ├── centerpoint_01voxel_second_secfpn_nus.py │ │ │ │ ├── centerpoint_02pillar_second_secfpn_nus.py │ │ │ │ ├── dgcnn.py │ │ │ │ ├── fcos3d.py │ │ │ │ ├── groupfree3d.py │ │ │ │ ├── h3dnet.py │ │ │ │ ├── hv_pointpillars_fpn_lyft.py │ │ │ │ ├── hv_pointpillars_fpn_nus.py │ │ │ │ ├── hv_pointpillars_fpn_range100_lyft.py │ │ │ │ ├── hv_pointpillars_secfpn_kitti.py │ │ │ │ ├── hv_pointpillars_secfpn_waymo.py │ │ │ │ ├── hv_second_secfpn_kitti.py │ │ │ │ ├── hv_second_secfpn_waymo.py │ │ │ │ ├── imvotenet_image.py │ │ │ │ ├── mask_rcnn_r50_fpn.py │ │ │ │ ├── paconv_cuda_ssg.py │ │ │ │ ├── paconv_ssg.py │ │ │ │ ├── parta2.py │ │ │ │ ├── pgd.py │ │ │ │ ├── point_rcnn.py │ │ │ │ ├── pointnet2_msg.py │ │ │ │ ├── pointnet2_ssg.py │ │ │ │ ├── smoke.py │ │ │ │ └── votenet.py │ │ │ └── schedules │ │ │ │ ├── cosine.py │ │ │ │ ├── cyclic_20e.py │ │ │ │ ├── cyclic_40e.py │ │ │ │ ├── mmdet_schedule_1x.py │ │ │ │ ├── schedule_2x.py │ │ │ │ ├── schedule_3x.py │ │ │ │ ├── seg_cosine_100e.py │ │ │ │ ├── seg_cosine_150e.py │ │ │ │ ├── seg_cosine_200e.py │ │ │ │ └── seg_cosine_50e.py │ │ ├── benchmark │ │ │ ├── hv_PartA2_secfpn_4x8_cyclic_80e_pcdet_kitti-3d-3class.py │ │ │ ├── hv_pointpillars_secfpn_3x8_100e_det3d_kitti-3d-car.py │ │ │ ├── hv_pointpillars_secfpn_4x8_80e_pcdet_kitti-3d-3class.py │ │ │ └── hv_second_secfpn_4x8_80e_pcdet_kitti-3d-3class.py │ │ ├── centerpoint │ │ │ ├── README.md │ │ │ ├── centerpoint_0075voxel_second_secfpn_4x8_cyclic_20e_nus.py │ │ │ ├── centerpoint_0075voxel_second_secfpn_circlenms_4x8_cyclic_20e_nus.py │ │ │ ├── centerpoint_0075voxel_second_secfpn_dcn_4x8_cyclic_20e_nus.py │ │ │ ├── centerpoint_0075voxel_second_secfpn_dcn_4x8_cyclic_flip-tta_20e_nus.py │ │ │ ├── centerpoint_0075voxel_second_secfpn_dcn_4x8_cyclic_tta_20e_nus.py │ │ │ ├── centerpoint_0075voxel_second_secfpn_dcn_circlenms_4x8_cyclic_20e_nus.py │ │ │ ├── centerpoint_0075voxel_second_secfpn_dcn_circlenms_4x8_cyclic_flip-tta_20e_nus.py │ │ │ ├── centerpoint_01voxel_second_secfpn_4x8_cyclic_20e_nus.py │ │ │ ├── centerpoint_01voxel_second_secfpn_circlenms_4x8_cyclic_20e_nus.py │ │ │ ├── centerpoint_01voxel_second_secfpn_dcn_4x8_cyclic_20e_nus.py │ │ │ ├── centerpoint_01voxel_second_secfpn_dcn_circlenms_4x8_cyclic_20e_nus.py │ │ │ ├── centerpoint_02pillar_second_secfpn_4x8_cyclic_20e_nus.py │ │ │ ├── centerpoint_02pillar_second_secfpn_circlenms_4x8_cyclic_20e_nus.py │ │ │ ├── centerpoint_02pillar_second_secfpn_dcn_4x8_cyclic_20e_nus.py │ │ │ ├── centerpoint_02pillar_second_secfpn_dcn_circlenms_4x8_cyclic_20e_nus.py │ │ │ └── metafile.yml │ │ ├── dgcnn │ │ │ ├── README.md │ │ │ ├── dgcnn_32x4_cosine_100e_s3dis_seg-3d-13class.py │ │ │ └── metafile.yml │ │ ├── dynamic_voxelization │ │ │ ├── README.md │ │ │ ├── dv_pointpillars_secfpn_6x8_160e_kitti-3d-car.py │ │ │ ├── dv_second_secfpn_2x8_cosine_80e_kitti-3d-3class.py │ │ │ ├── dv_second_secfpn_6x8_80e_kitti-3d-car.py │ │ │ └── metafile.yml │ │ ├── fcos3d │ │ │ ├── README.md │ │ │ ├── fcos3d_r101_caffe_fpn_gn-head_dcn_2x8_1x_nus-mono3d.py │ │ │ ├── fcos3d_r101_caffe_fpn_gn-head_dcn_2x8_1x_nus-mono3d_finetune.py │ │ │ └── metafile.yml │ │ ├── free_anchor │ │ │ ├── README.md │ │ │ ├── hv_pointpillars_fpn_sbn-all_free-anchor_4x8_2x_nus-3d.py │ │ │ ├── hv_pointpillars_regnet-1.6gf_fpn_sbn-all_free-anchor_4x8_2x_nus-3d.py │ │ │ ├── hv_pointpillars_regnet-1.6gf_fpn_sbn-all_free-anchor_strong-aug_4x8_3x_nus-3d.py │ │ │ ├── hv_pointpillars_regnet-3.2gf_fpn_sbn-all_free-anchor_4x8_2x_nus-3d.py │ │ │ ├── hv_pointpillars_regnet-3.2gf_fpn_sbn-all_free-anchor_strong-aug_4x8_3x_nus-3d.py │ │ │ ├── hv_pointpillars_regnet-400mf_fpn_sbn-all_free-anchor_4x8_2x_nus-3d.py │ │ │ └── metafile.yml │ │ ├── groupfree3d │ │ │ ├── README.md │ │ │ ├── groupfree3d_8x4_scannet-3d-18class-L12-O256.py │ │ │ ├── groupfree3d_8x4_scannet-3d-18class-L6-O256.py │ │ │ ├── groupfree3d_8x4_scannet-3d-18class-w2x-L12-O256.py │ │ │ ├── groupfree3d_8x4_scannet-3d-18class-w2x-L12-O512.py │ │ │ └── metafile.yml │ │ ├── h3dnet │ │ │ ├── README.md │ │ │ ├── h3dnet_3x8_scannet-3d-18class.py │ │ │ └── metafile.yml │ │ ├── imvotenet │ │ │ ├── README.md │ │ │ ├── imvotenet_faster_rcnn_r50_fpn_2x4_sunrgbd-3d-10class.py │ │ │ ├── imvotenet_stage2_16x8_sunrgbd-3d-10class.py │ │ │ └── metafile.yml │ │ ├── imvoxelnet │ │ │ ├── README.md │ │ │ ├── imvoxelnet_4x8_kitti-3d-car.py │ │ │ └── metafile.yml │ │ ├── monoflex │ │ │ ├── README.md │ │ │ └── metafile.yml │ │ ├── mvxnet │ │ │ ├── README.md │ │ │ ├── dv_mvx-fpn_second_secfpn_adamw_2x8_80e_kitti-3d-3class.py │ │ │ └── metafile.yml │ │ ├── nuimages │ │ │ ├── README.md │ │ │ ├── cascade_mask_rcnn_r101_fpn_1x_nuim.py │ │ │ ├── cascade_mask_rcnn_r50_fpn_1x_nuim.py │ │ │ ├── cascade_mask_rcnn_r50_fpn_coco-20e_1x_nuim.py │ │ │ ├── cascade_mask_rcnn_r50_fpn_coco-20e_20e_nuim.py │ │ │ ├── cascade_mask_rcnn_x101_32x4d_fpn_1x_nuim.py │ │ │ ├── htc_r50_fpn_1x_nuim.py │ │ │ ├── htc_r50_fpn_coco-20e_1x_nuim.py │ │ │ ├── htc_r50_fpn_coco-20e_20e_nuim.py │ │ │ ├── htc_without_semantic_r50_fpn_1x_nuim.py │ │ │ ├── htc_x101_64x4d_fpn_dconv_c3-c5_coco-20e_16x1_20e_nuim.py │ │ │ ├── mask_rcnn_r101_fpn_1x_nuim.py │ │ │ ├── mask_rcnn_r50_caffe_fpn_1x_nuim.py │ │ │ ├── mask_rcnn_r50_caffe_fpn_coco-3x_1x_nuim.py │ │ │ ├── mask_rcnn_r50_caffe_fpn_coco-3x_20e_nuim.py │ │ │ ├── mask_rcnn_r50_fpn_1x_nuim.py │ │ │ ├── mask_rcnn_r50_fpn_coco-2x_1x_nuim.py │ │ │ ├── mask_rcnn_r50_fpn_coco-2x_1x_nus-2d.py │ │ │ ├── mask_rcnn_x101_32x4d_fpn_1x_nuim.py │ │ │ └── metafile.yml │ │ ├── paconv │ │ │ ├── README.md │ │ │ ├── metafile.yml │ │ │ ├── paconv_cuda_ssg_8x8_cosine_200e_s3dis_seg-3d-13class.py │ │ │ └── paconv_ssg_8x8_cosine_150e_s3dis_seg-3d-13class.py │ │ ├── parta2 │ │ │ ├── README.md │ │ │ ├── hv_PartA2_secfpn_2x8_cyclic_80e_kitti-3d-3class.py │ │ │ ├── hv_PartA2_secfpn_2x8_cyclic_80e_kitti-3d-car.py │ │ │ └── metafile.yml │ │ ├── pgd │ │ │ ├── README.md │ │ │ ├── metafile.yml │ │ │ ├── pgd_r101_caffe_fpn_gn-head_2x16_1x_nus-mono3d.py │ │ │ ├── pgd_r101_caffe_fpn_gn-head_2x16_1x_nus-mono3d_finetune.py │ │ │ ├── pgd_r101_caffe_fpn_gn-head_2x16_2x_nus-mono3d.py │ │ │ ├── pgd_r101_caffe_fpn_gn-head_2x16_2x_nus-mono3d_finetune.py │ │ │ ├── pgd_r101_caffe_fpn_gn-head_3x4_4x_kitti-mono3d.py │ │ │ └── pgd_r101_caffe_fpn_gn-head_4x8_1x_nus-mono3d_res256x704.py │ │ ├── point_rcnn │ │ │ ├── README.md │ │ │ ├── metafile.yml │ │ │ └── point_rcnn_2x8_kitti-3d-3classes.py │ │ ├── pointnet2 │ │ │ ├── README.md │ │ │ ├── metafile.yml │ │ │ ├── pointnet2_msg_16x2_cosine_250e_scannet_seg-3d-20class.py │ │ │ ├── pointnet2_msg_16x2_cosine_80e_s3dis_seg-3d-13class.py │ │ │ ├── pointnet2_msg_xyz-only_16x2_cosine_250e_scannet_seg-3d-20class.py │ │ │ ├── pointnet2_ssg_16x2_cosine_200e_scannet_seg-3d-20class.py │ │ │ ├── pointnet2_ssg_16x2_cosine_50e_s3dis_seg-3d-13class.py │ │ │ └── pointnet2_ssg_xyz-only_16x2_cosine_200e_scannet_seg-3d-20class.py │ │ ├── pointpillars │ │ │ ├── README.md │ │ │ ├── hv_pointpillars_fpn_sbn-all_2x8_2x_lyft-3d.py │ │ │ ├── hv_pointpillars_fpn_sbn-all_4x8_2x_nus-3d.py │ │ │ ├── hv_pointpillars_fpn_sbn-all_fp16_2x8_2x_nus-3d.py │ │ │ ├── hv_pointpillars_fpn_sbn-all_range100_2x8_2x_lyft-3d.py │ │ │ ├── hv_pointpillars_secfpn_6x8_160e_kitti-3d-3class.py │ │ │ ├── hv_pointpillars_secfpn_6x8_160e_kitti-3d-car.py │ │ │ ├── hv_pointpillars_secfpn_sbn-all_2x8_2x_lyft-3d.py │ │ │ ├── hv_pointpillars_secfpn_sbn-all_4x8_2x_nus-3d.py │ │ │ ├── hv_pointpillars_secfpn_sbn-all_fp16_2x8_2x_nus-3d.py │ │ │ ├── hv_pointpillars_secfpn_sbn-all_range100_2x8_2x_lyft-3d.py │ │ │ ├── hv_pointpillars_secfpn_sbn_2x16_2x_waymo-3d-3class.py │ │ │ ├── hv_pointpillars_secfpn_sbn_2x16_2x_waymo-3d-car.py │ │ │ ├── hv_pointpillars_secfpn_sbn_2x16_2x_waymoD5-3d-3class.py │ │ │ ├── hv_pointpillars_secfpn_sbn_2x16_2x_waymoD5-3d-3class_fix_order.py │ │ │ ├── hv_pointpillars_secfpn_sbn_2x16_2x_waymoD5-3d-car.py │ │ │ └── metafile.yml │ │ ├── regnet │ │ │ ├── README.md │ │ │ ├── hv_pointpillars_regnet-1.6gf_fpn_sbn-all_4x8_2x_nus-3d.py │ │ │ ├── hv_pointpillars_regnet-400mf_fpn_sbn-all_2x8_2x_lyft-3d.py │ │ │ ├── hv_pointpillars_regnet-400mf_fpn_sbn-all_4x8_2x_nus-3d.py │ │ │ ├── hv_pointpillars_regnet-400mf_fpn_sbn-all_fp16_2x8_2x_nus-3d.py │ │ │ ├── hv_pointpillars_regnet-400mf_fpn_sbn-all_range100_2x8_2x_lyft-3d.py │ │ │ ├── hv_pointpillars_regnet-400mf_secfpn_sbn-all_2x8_2x_lyft-3d.py │ │ │ ├── hv_pointpillars_regnet-400mf_secfpn_sbn-all_4x8_2x_nus-3d.py │ │ │ ├── hv_pointpillars_regnet-400mf_secfpn_sbn-all_range100_2x8_2x_lyft-3d.py │ │ │ └── metafile.yml │ │ ├── sassd │ │ │ ├── README.md │ │ │ └── sassd_6x8_80e_kitti-3d-3class.py │ │ ├── second │ │ │ ├── README.md │ │ │ ├── hv_second_secfpn_6x8_80e_kitti-3d-3class.py │ │ │ ├── hv_second_secfpn_6x8_80e_kitti-3d-car.py │ │ │ ├── hv_second_secfpn_fp16_6x8_80e_kitti-3d-3class.py │ │ │ ├── hv_second_secfpn_fp16_6x8_80e_kitti-3d-car.py │ │ │ ├── hv_second_secfpn_sbn_2x16_2x_waymoD5-3d-3class.py │ │ │ └── metafile.yml │ │ ├── smoke │ │ │ ├── README.md │ │ │ ├── metafile.yml │ │ │ └── smoke_dla34_pytorch_dlaneck_gn-all_8x4_6x_kitti-mono3d.py │ │ ├── ssn │ │ │ ├── README.md │ │ │ ├── hv_ssn_regnet-400mf_secfpn_sbn-all_1x16_2x_lyft-3d.py │ │ │ ├── hv_ssn_regnet-400mf_secfpn_sbn-all_2x16_2x_nus-3d.py │ │ │ ├── hv_ssn_secfpn_sbn-all_2x16_2x_lyft-3d.py │ │ │ ├── hv_ssn_secfpn_sbn-all_2x16_2x_nus-3d.py │ │ │ └── metafile.yml │ │ └── votenet │ │ │ ├── README.md │ │ │ ├── metafile.yml │ │ │ ├── votenet_16x8_sunrgbd-3d-10class.py │ │ │ ├── votenet_8x8_scannet-3d-18class.py │ │ │ └── votenet_iouloss_8x8_scannet-3d-18class.py │ ├── demo │ │ ├── mono_det_demo.py │ │ ├── multi_modality_demo.py │ │ ├── pc_seg_demo.py │ │ └── pcd_demo.py │ ├── docker │ │ ├── Dockerfile │ │ └── serve │ │ │ ├── Dockerfile │ │ │ ├── config.properties │ │ │ └── entrypoint.sh │ ├── docs │ │ ├── en │ │ │ ├── 1_exist_data_model.md │ │ │ ├── 2_new_data_model.md │ │ │ ├── Makefile │ │ │ ├── _static │ │ │ │ └── css │ │ │ │ │ └── readthedocs.css │ │ │ ├── api.rst │ │ │ ├── benchmarks.md │ │ │ ├── changelog.md │ │ │ ├── compatibility.md │ │ │ ├── conf.py │ │ │ ├── data_preparation.md │ │ │ ├── datasets │ │ │ │ ├── index.rst │ │ │ │ ├── kitti_det.md │ │ │ │ ├── lyft_det.md │ │ │ │ ├── nuscenes_det.md │ │ │ │ ├── s3dis_sem_seg.md │ │ │ │ ├── scannet_det.md │ │ │ │ ├── scannet_sem_seg.md │ │ │ │ ├── sunrgbd_det.md │ │ │ │ └── waymo_det.md │ │ │ ├── demo.md │ │ │ ├── faq.md │ │ │ ├── getting_started.md │ │ │ ├── index.rst │ │ │ ├── make.bat │ │ │ ├── model_zoo.md │ │ │ ├── stat.py │ │ │ ├── supported_tasks │ │ │ │ ├── index.rst │ │ │ │ ├── lidar_det3d.md │ │ │ │ ├── lidar_sem_seg3d.md │ │ │ │ └── vision_det3d.md │ │ │ ├── switch_language.md │ │ │ ├── tutorials │ │ │ │ ├── backends_support.md │ │ │ │ ├── config.md │ │ │ │ ├── coord_sys_tutorial.md │ │ │ │ ├── customize_dataset.md │ │ │ │ ├── customize_models.md │ │ │ │ ├── customize_runtime.md │ │ │ │ ├── data_pipeline.md │ │ │ │ ├── index.rst │ │ │ │ ├── model_deployment.md │ │ │ │ └── pure_point_cloud_dataset.md │ │ │ └── useful_tools.md │ │ └── zh_cn │ │ │ ├── 1_exist_data_model.md │ │ │ ├── 2_new_data_model.md │ │ │ ├── Makefile │ │ │ ├── _static │ │ │ └── css │ │ │ │ └── readthedocs.css │ │ │ ├── api.rst │ │ │ ├── benchmarks.md │ │ │ ├── changelog.md │ │ │ ├── compatibility.md │ │ │ ├── conf.py │ │ │ ├── data_preparation.md │ │ │ ├── datasets │ │ │ ├── index.rst │ │ │ ├── kitti_det.md │ │ │ ├── lyft_det.md │ │ │ ├── nuscenes_det.md │ │ │ ├── s3dis_sem_seg.md │ │ │ ├── scannet_det.md │ │ │ ├── scannet_sem_seg.md │ │ │ ├── sunrgbd_det.md │ │ │ └── waymo_det.md │ │ │ ├── demo.md │ │ │ ├── faq.md │ │ │ ├── getting_started.md │ │ │ ├── index.rst │ │ │ ├── make.bat │ │ │ ├── model_zoo.md │ │ │ ├── stat.py │ │ │ ├── supported_tasks │ │ │ ├── index.rst │ │ │ ├── lidar_det3d.md │ │ │ ├── lidar_sem_seg3d.md │ │ │ └── vision_det3d.md │ │ │ ├── switch_language.md │ │ │ ├── tutorials │ │ │ ├── backends_support.md │ │ │ ├── config.md │ │ │ ├── coord_sys_tutorial.md │ │ │ ├── customize_dataset.md │ │ │ ├── customize_models.md │ │ │ ├── customize_runtime.md │ │ │ ├── data_pipeline.md │ │ │ ├── index.rst │ │ │ └── model_deployment.md │ │ │ └── useful_tools.md │ ├── mmdet3d │ │ ├── __init__.py │ │ ├── apis │ │ │ ├── __init__.py │ │ │ ├── inference.py │ │ │ ├── test.py │ │ │ └── train.py │ │ ├── core │ │ │ ├── __init__.py │ │ │ ├── anchor │ │ │ │ ├── __init__.py │ │ │ │ └── anchor_3d_generator.py │ │ │ ├── bbox │ │ │ │ ├── __init__.py │ │ │ │ ├── assigners │ │ │ │ │ └── __init__.py │ │ │ │ ├── box_np_ops.py │ │ │ │ ├── coders │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── anchor_free_bbox_coder.py │ │ │ │ │ ├── centerpoint_bbox_coders.py │ │ │ │ │ ├── delta_xyzwhlr_bbox_coder.py │ │ │ │ │ ├── fcos3d_bbox_coder.py │ │ │ │ │ ├── groupfree3d_bbox_coder.py │ │ │ │ │ ├── monoflex_bbox_coder.py │ │ │ │ │ ├── partial_bin_based_bbox_coder.py │ │ │ │ │ ├── pgd_bbox_coder.py │ │ │ │ │ ├── point_xyzwhlr_bbox_coder.py │ │ │ │ │ └── smoke_bbox_coder.py │ │ │ │ ├── iou_calculators │ │ │ │ │ ├── __init__.py │ │ │ │ │ └── iou3d_calculator.py │ │ │ │ ├── samplers │ │ │ │ │ ├── __init__.py │ │ │ │ │ └── iou_neg_piecewise_sampler.py │ │ │ │ ├── structures │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── base_box3d.py │ │ │ │ │ ├── box_3d_mode.py │ │ │ │ │ ├── cam_box3d.py │ │ │ │ │ ├── coord_3d_mode.py │ │ │ │ │ ├── depth_box3d.py │ │ │ │ │ ├── lidar_box3d.py │ │ │ │ │ └── utils.py │ │ │ │ └── transforms.py │ │ │ ├── evaluation │ │ │ │ ├── __init__.py │ │ │ │ ├── indoor_eval.py │ │ │ │ ├── instance_seg_eval.py │ │ │ │ ├── kitti_utils │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── eval.py │ │ │ │ │ └── rotate_iou.py │ │ │ │ ├── lyft_eval.py │ │ │ │ ├── scannet_utils │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── evaluate_semantic_instance.py │ │ │ │ │ └── util_3d.py │ │ │ │ ├── seg_eval.py │ │ │ │ └── waymo_utils │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── compute_detection_let_metrics_main │ │ │ │ │ └── prediction_kitti_to_waymo.py │ │ │ ├── points │ │ │ │ ├── __init__.py │ │ │ │ ├── base_points.py │ │ │ │ ├── cam_points.py │ │ │ │ ├── depth_points.py │ │ │ │ └── lidar_points.py │ │ │ ├── post_processing │ │ │ │ ├── __init__.py │ │ │ │ ├── box3d_nms.py │ │ │ │ └── merge_augs.py │ │ │ ├── utils │ │ │ │ ├── __init__.py │ │ │ │ ├── array_converter.py │ │ │ │ └── gaussian.py │ │ │ ├── visualizer │ │ │ │ ├── __init__.py │ │ │ │ ├── image_vis.py │ │ │ │ ├── open3d_vis.py │ │ │ │ └── show_result.py │ │ │ └── voxel │ │ │ │ ├── __init__.py │ │ │ │ ├── builder.py │ │ │ │ └── voxel_generator.py │ │ ├── datasets │ │ │ ├── __init__.py │ │ │ ├── builder.py │ │ │ ├── custom_3d.py │ │ │ ├── custom_3d_seg.py │ │ │ ├── dataset_wrappers.py │ │ │ ├── kitti2d_dataset.py │ │ │ ├── kitti_dataset.py │ │ │ ├── kitti_mono_dataset.py │ │ │ ├── lyft_dataset.py │ │ │ ├── nuscenes_dataset.py │ │ │ ├── nuscenes_mono_dataset.py │ │ │ ├── pipelines │ │ │ │ ├── __init__.py │ │ │ │ ├── compose.py │ │ │ │ ├── data_augment_utils.py │ │ │ │ ├── dbsampler.py │ │ │ │ ├── formating.py │ │ │ │ ├── loading.py │ │ │ │ ├── test_time_aug.py │ │ │ │ └── transforms_3d.py │ │ │ ├── s3dis_dataset.py │ │ │ ├── scannet_dataset.py │ │ │ ├── semantickitti_dataset.py │ │ │ ├── sunrgbd_dataset.py │ │ │ ├── utils.py │ │ │ └── waymo_dataset.py │ │ ├── models │ │ │ ├── __init__.py │ │ │ ├── backbones │ │ │ │ ├── __init__.py │ │ │ │ ├── base_pointnet.py │ │ │ │ ├── dgcnn.py │ │ │ │ ├── dla.py │ │ │ │ ├── mink_resnet.py │ │ │ │ ├── multi_backbone.py │ │ │ │ ├── nostem_regnet.py │ │ │ │ ├── pointnet2_sa_msg.py │ │ │ │ ├── pointnet2_sa_ssg.py │ │ │ │ └── second.py │ │ │ ├── builder.py │ │ │ ├── decode_heads │ │ │ │ ├── __init__.py │ │ │ │ ├── decode_head.py │ │ │ │ ├── dgcnn_head.py │ │ │ │ ├── paconv_head.py │ │ │ │ └── pointnet2_head.py │ │ │ ├── dense_heads │ │ │ │ ├── __init__.py │ │ │ │ ├── anchor3d_head.py │ │ │ │ ├── anchor_free_mono3d_head.py │ │ │ │ ├── base_conv_bbox_head.py │ │ │ │ ├── base_mono3d_dense_head.py │ │ │ │ ├── centerpoint_head.py │ │ │ │ ├── fcos_mono3d_head.py │ │ │ │ ├── free_anchor3d_head.py │ │ │ │ ├── groupfree3d_head.py │ │ │ │ ├── monoflex_head.py │ │ │ │ ├── parta2_rpn_head.py │ │ │ │ ├── pgd_head.py │ │ │ │ ├── point_rpn_head.py │ │ │ │ ├── shape_aware_head.py │ │ │ │ ├── smoke_mono3d_head.py │ │ │ │ ├── ssd_3d_head.py │ │ │ │ ├── train_mixins.py │ │ │ │ └── vote_head.py │ │ │ ├── detectors │ │ │ │ ├── __init__.py │ │ │ │ ├── base.py │ │ │ │ ├── centerpoint.py │ │ │ │ ├── dynamic_voxelnet.py │ │ │ │ ├── fcos_mono3d.py │ │ │ │ ├── groupfree3dnet.py │ │ │ │ ├── h3dnet.py │ │ │ │ ├── imvotenet.py │ │ │ │ ├── imvoxelnet.py │ │ │ │ ├── mvx_faster_rcnn.py │ │ │ │ ├── mvx_two_stage.py │ │ │ │ ├── parta2.py │ │ │ │ ├── point_rcnn.py │ │ │ │ ├── sassd.py │ │ │ │ ├── single_stage.py │ │ │ │ ├── single_stage_mono3d.py │ │ │ │ ├── smoke_mono3d.py │ │ │ │ ├── ssd3dnet.py │ │ │ │ ├── two_stage.py │ │ │ │ ├── votenet.py │ │ │ │ └── voxelnet.py │ │ │ ├── fusion_layers │ │ │ │ ├── __init__.py │ │ │ │ ├── coord_transform.py │ │ │ │ ├── point_fusion.py │ │ │ │ └── vote_fusion.py │ │ │ ├── losses │ │ │ │ ├── __init__.py │ │ │ │ ├── axis_aligned_iou_loss.py │ │ │ │ ├── chamfer_distance.py │ │ │ │ ├── multibin_loss.py │ │ │ │ ├── paconv_regularization_loss.py │ │ │ │ └── uncertain_smooth_l1_loss.py │ │ │ ├── middle_encoders │ │ │ │ ├── __init__.py │ │ │ │ ├── pillar_scatter.py │ │ │ │ ├── sparse_encoder.py │ │ │ │ └── sparse_unet.py │ │ │ ├── model_utils │ │ │ │ ├── __init__.py │ │ │ │ ├── edge_fusion_module.py │ │ │ │ ├── transformer.py │ │ │ │ └── vote_module.py │ │ │ ├── necks │ │ │ │ ├── __init__.py │ │ │ │ ├── dla_neck.py │ │ │ │ ├── imvoxel_neck.py │ │ │ │ ├── pointnet2_fp_neck.py │ │ │ │ └── second_fpn.py │ │ │ ├── roi_heads │ │ │ │ ├── __init__.py │ │ │ │ ├── base_3droi_head.py │ │ │ │ ├── bbox_heads │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── h3d_bbox_head.py │ │ │ │ │ ├── parta2_bbox_head.py │ │ │ │ │ └── point_rcnn_bbox_head.py │ │ │ │ ├── h3d_roi_head.py │ │ │ │ ├── mask_heads │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── pointwise_semantic_head.py │ │ │ │ │ └── primitive_head.py │ │ │ │ ├── part_aggregation_roi_head.py │ │ │ │ ├── point_rcnn_roi_head.py │ │ │ │ └── roi_extractors │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── single_roiaware_extractor.py │ │ │ │ │ └── single_roipoint_extractor.py │ │ │ ├── segmentors │ │ │ │ ├── __init__.py │ │ │ │ ├── base.py │ │ │ │ └── encoder_decoder.py │ │ │ ├── utils │ │ │ │ ├── __init__.py │ │ │ │ ├── clip_sigmoid.py │ │ │ │ ├── edge_indices.py │ │ │ │ ├── gen_keypoints.py │ │ │ │ ├── handle_objs.py │ │ │ │ └── mlp.py │ │ │ └── voxel_encoders │ │ │ │ ├── __init__.py │ │ │ │ ├── pillar_encoder.py │ │ │ │ ├── utils.py │ │ │ │ └── voxel_encoder.py │ │ ├── ops │ │ │ ├── __init__.py │ │ │ ├── dgcnn_modules │ │ │ │ ├── __init__.py │ │ │ │ ├── dgcnn_fa_module.py │ │ │ │ ├── dgcnn_fp_module.py │ │ │ │ └── dgcnn_gf_module.py │ │ │ ├── norm.py │ │ │ ├── paconv │ │ │ │ ├── __init__.py │ │ │ │ ├── paconv.py │ │ │ │ └── utils.py │ │ │ ├── pointnet_modules │ │ │ │ ├── __init__.py │ │ │ │ ├── builder.py │ │ │ │ ├── paconv_sa_module.py │ │ │ │ ├── point_fp_module.py │ │ │ │ └── point_sa_module.py │ │ │ ├── sparse_block.py │ │ │ └── spconv │ │ │ │ ├── __init__.py │ │ │ │ └── overwrite_spconv │ │ │ │ ├── __init__.py │ │ │ │ └── write_spconv2.py │ │ ├── utils │ │ │ ├── __init__.py │ │ │ ├── collect_env.py │ │ │ ├── compat_cfg.py │ │ │ ├── logger.py │ │ │ ├── misc.py │ │ │ └── setup_env.py │ │ └── version.py │ ├── model-index.yml │ ├── requirements.txt │ ├── requirements │ │ ├── build.txt │ │ ├── docs.txt │ │ ├── mminstall.txt │ │ ├── optional.txt │ │ ├── readthedocs.txt │ │ ├── runtime.txt │ │ └── tests.txt │ ├── resources │ │ ├── mmdet3d_outdoor_demo.gif │ │ ├── nuimages_demo.gif │ │ └── open3d_visual.gif │ ├── setup.cfg │ ├── setup.py │ ├── test_waymo.py │ ├── tests │ │ ├── test_data │ │ │ ├── test_datasets │ │ │ │ ├── test_dataset_wrappers.py │ │ │ │ ├── test_kitti_dataset.py │ │ │ │ ├── test_kitti_mono_dataset.py │ │ │ │ ├── test_lyft_dataset.py │ │ │ │ ├── test_nuscene_dataset.py │ │ │ │ ├── test_nuscenes_mono_dataset.py │ │ │ │ ├── test_s3dis_dataset.py │ │ │ │ ├── test_scannet_dataset.py │ │ │ │ ├── test_semantickitti_dataset.py │ │ │ │ ├── test_sunrgbd_dataset.py │ │ │ │ └── test_waymo_dataset.py │ │ │ └── test_pipelines │ │ │ │ ├── test_augmentations │ │ │ │ ├── test_data_augment_utils.py │ │ │ │ ├── test_test_augment_utils.py │ │ │ │ └── test_transforms_3d.py │ │ │ │ ├── test_indoor_pipeline.py │ │ │ │ ├── test_indoor_sample.py │ │ │ │ ├── test_loadings │ │ │ │ ├── test_load_images_from_multi_views.py │ │ │ │ ├── test_load_points_from_multi_sweeps.py │ │ │ │ └── test_loading.py │ │ │ │ └── test_outdoor_pipeline.py │ │ ├── test_metrics │ │ │ ├── test_indoor_eval.py │ │ │ ├── test_instance_seg_eval.py │ │ │ ├── test_kitti_eval.py │ │ │ ├── test_losses.py │ │ │ └── test_seg_eval.py │ │ ├── test_models │ │ │ ├── test_backbones.py │ │ │ ├── test_common_modules │ │ │ │ ├── test_dgcnn_modules.py │ │ │ │ ├── test_middle_encoders.py │ │ │ │ ├── test_paconv_modules.py │ │ │ │ ├── test_paconv_ops.py │ │ │ │ ├── test_pointnet_modules.py │ │ │ │ ├── test_sparse_unet.py │ │ │ │ └── test_vote_module.py │ │ │ ├── test_detectors.py │ │ │ ├── test_forward.py │ │ │ ├── test_fusion │ │ │ │ ├── test_fusion_coord_trans.py │ │ │ │ ├── test_point_fusion.py │ │ │ │ └── test_vote_fusion.py │ │ │ ├── test_heads │ │ │ │ ├── test_dgcnn_decode_head.py │ │ │ │ ├── test_heads.py │ │ │ │ ├── test_paconv_decode_head.py │ │ │ │ ├── test_parta2_bbox_head.py │ │ │ │ ├── test_pointnet2_decode_head.py │ │ │ │ ├── test_roi_extractors.py │ │ │ │ └── test_semantic_heads.py │ │ │ ├── test_necks │ │ │ │ ├── test_fpn.py │ │ │ │ └── test_necks.py │ │ │ ├── test_segmentors.py │ │ │ └── test_voxel_encoder │ │ │ │ ├── test_voxel_encoders.py │ │ │ │ └── test_voxel_generator.py │ │ ├── test_runtime │ │ │ ├── test_apis.py │ │ │ └── test_config.py │ │ ├── test_samples │ │ │ └── parta2_roihead_inputs.npz │ │ └── test_utils │ │ │ ├── test_anchors.py │ │ │ ├── test_assigners.py │ │ │ ├── test_bbox_coders.py │ │ │ ├── test_box3d.py │ │ │ ├── test_box_np_ops.py │ │ │ ├── test_compat_cfg.py │ │ │ ├── test_coord_3d_mode.py │ │ │ ├── test_merge_augs.py │ │ │ ├── test_nms.py │ │ │ ├── test_points.py │ │ │ ├── test_samplers.py │ │ │ ├── test_setup_env.py │ │ │ └── test_utils.py │ └── tools │ │ ├── analysis_tools │ │ ├── analyze_logs.py │ │ ├── benchmark.py │ │ └── get_flops.py │ │ ├── create_data.py │ │ ├── create_data.sh │ │ ├── data_converter │ │ ├── __init__.py │ │ ├── create_gt_database.py │ │ ├── indoor_converter.py │ │ ├── kitti_converter.py │ │ ├── kitti_data_utils.py │ │ ├── lyft_converter.py │ │ ├── lyft_data_fixer.py │ │ ├── nuimage_converter.py │ │ ├── nuscenes_converter.py │ │ ├── s3dis_data_utils.py │ │ ├── scannet_data_utils.py │ │ ├── sunrgbd_data_utils.py │ │ └── waymo_converter.py │ │ ├── deployment │ │ ├── mmdet3d2torchserve.py │ │ ├── mmdet3d_handler.py │ │ └── test_torchserver.py │ │ ├── dist_test.sh │ │ ├── dist_train.sh │ │ ├── misc │ │ ├── browse_dataset.py │ │ ├── fuse_conv_bn.py │ │ ├── print_config.py │ │ └── visualize_results.py │ │ ├── model_converters │ │ ├── convert_h3dnet_checkpoints.py │ │ ├── convert_votenet_checkpoints.py │ │ ├── publish_model.py │ │ └── regnet2mmdet.py │ │ ├── slurm_test.sh │ │ ├── slurm_train.sh │ │ ├── test.py │ │ ├── train.py │ │ ├── update_data_coords.py │ │ └── update_data_coords.sh ├── mmseg │ ├── .circleci │ │ └── config.yml │ ├── .dev │ │ ├── batch_test_list.py │ │ ├── batch_train_list.txt │ │ ├── benchmark_evaluation.sh │ │ ├── benchmark_inference.py │ │ ├── benchmark_train.sh │ │ ├── check_urls.py │ │ ├── gather_benchmark_evaluation_results.py │ │ ├── gather_benchmark_train_results.py │ │ ├── gather_models.py │ │ ├── generate_benchmark_evaluation_script.py │ │ ├── generate_benchmark_train_script.py │ │ ├── log_collector │ │ │ ├── example_config.py │ │ │ ├── log_collector.py │ │ │ ├── readme.md │ │ │ └── utils.py │ │ ├── md2yml.py │ │ └── upload_modelzoo.py │ ├── .pre-commit-config.yaml │ ├── .readthedocs.yml │ ├── CITATION.cff │ ├── LICENSE │ ├── MANIFEST.in │ ├── README.md │ ├── README_zh-CN.md │ ├── configs │ │ ├── _base_ │ │ │ ├── datasets │ │ │ │ ├── ade20k.py │ │ │ │ ├── ade20k_640x640.py │ │ │ │ ├── chase_db1.py │ │ │ │ ├── cityscapes.py │ │ │ │ ├── cityscapes_1024x1024.py │ │ │ │ ├── cityscapes_768x768.py │ │ │ │ ├── cityscapes_769x769.py │ │ │ │ ├── cityscapes_832x832.py │ │ │ │ ├── coco-stuff10k.py │ │ │ │ ├── coco-stuff164k.py │ │ │ │ ├── drive.py │ │ │ │ ├── hrf.py │ │ │ │ ├── isaid.py │ │ │ │ ├── loveda.py │ │ │ │ ├── pascal_context.py │ │ │ │ ├── pascal_context_59.py │ │ │ │ ├── pascal_voc12.py │ │ │ │ ├── pascal_voc12_aug.py │ │ │ │ ├── potsdam.py │ │ │ │ ├── stare.py │ │ │ │ └── vaihingen.py │ │ │ ├── default_runtime.py │ │ │ ├── models │ │ │ │ ├── ann_r50-d8.py │ │ │ │ ├── apcnet_r50-d8.py │ │ │ │ ├── bisenetv1_r18-d32.py │ │ │ │ ├── bisenetv2.py │ │ │ │ ├── ccnet_r50-d8.py │ │ │ │ ├── cgnet.py │ │ │ │ ├── danet_r50-d8.py │ │ │ │ ├── deeplabv3_r50-d8.py │ │ │ │ ├── deeplabv3_unet_s5-d16.py │ │ │ │ ├── deeplabv3plus_r50-d8.py │ │ │ │ ├── dmnet_r50-d8.py │ │ │ │ ├── dnl_r50-d8.py │ │ │ │ ├── dpt_vit-b16.py │ │ │ │ ├── emanet_r50-d8.py │ │ │ │ ├── encnet_r50-d8.py │ │ │ │ ├── erfnet_fcn.py │ │ │ │ ├── fast_scnn.py │ │ │ │ ├── fastfcn_r50-d32_jpu_psp.py │ │ │ │ ├── fcn_hr18.py │ │ │ │ ├── fcn_r50-d8.py │ │ │ │ ├── fcn_unet_s5-d16.py │ │ │ │ ├── fpn_r50.py │ │ │ │ ├── gcnet_r50-d8.py │ │ │ │ ├── icnet_r50-d8.py │ │ │ │ ├── isanet_r50-d8.py │ │ │ │ ├── lraspp_m-v3-d8.py │ │ │ │ ├── nonlocal_r50-d8.py │ │ │ │ ├── ocrnet_hr18.py │ │ │ │ ├── ocrnet_r50-d8.py │ │ │ │ ├── pointrend_r50.py │ │ │ │ ├── psanet_r50-d8.py │ │ │ │ ├── pspnet_r50-d8.py │ │ │ │ ├── pspnet_unet_s5-d16.py │ │ │ │ ├── segformer_mit-b0.py │ │ │ │ ├── segmenter_vit-b16_mask.py │ │ │ │ ├── setr_mla.py │ │ │ │ ├── setr_naive.py │ │ │ │ ├── setr_pup.py │ │ │ │ ├── stdc.py │ │ │ │ ├── twins_pcpvt-s_fpn.py │ │ │ │ ├── twins_pcpvt-s_upernet.py │ │ │ │ ├── upernet_convnext.py │ │ │ │ ├── upernet_r50.py │ │ │ │ ├── upernet_swin.py │ │ │ │ └── upernet_vit-b16_ln_mln.py │ │ │ └── schedules │ │ │ │ ├── schedule_160k.py │ │ │ │ ├── schedule_20k.py │ │ │ │ ├── schedule_320k.py │ │ │ │ ├── schedule_40k.py │ │ │ │ └── schedule_80k.py │ │ ├── ann │ │ │ ├── README.md │ │ │ ├── ann.yml │ │ │ ├── ann_r101-d8_512x1024_40k_cityscapes.py │ │ │ ├── ann_r101-d8_512x1024_80k_cityscapes.py │ │ │ ├── ann_r101-d8_512x512_160k_ade20k.py │ │ │ ├── ann_r101-d8_512x512_20k_voc12aug.py │ │ │ ├── ann_r101-d8_512x512_40k_voc12aug.py │ │ │ ├── ann_r101-d8_512x512_80k_ade20k.py │ │ │ ├── ann_r101-d8_769x769_40k_cityscapes.py │ │ │ ├── ann_r101-d8_769x769_80k_cityscapes.py │ │ │ ├── ann_r50-d8_512x1024_40k_cityscapes.py │ │ │ ├── ann_r50-d8_512x1024_80k_cityscapes.py │ │ │ ├── ann_r50-d8_512x512_160k_ade20k.py │ │ │ ├── ann_r50-d8_512x512_20k_voc12aug.py │ │ │ ├── ann_r50-d8_512x512_40k_voc12aug.py │ │ │ ├── ann_r50-d8_512x512_80k_ade20k.py │ │ │ ├── ann_r50-d8_769x769_40k_cityscapes.py │ │ │ └── ann_r50-d8_769x769_80k_cityscapes.py │ │ ├── apcnet │ │ │ ├── README.md │ │ │ ├── apcnet.yml │ │ │ ├── apcnet_r101-d8_512x1024_40k_cityscapes.py │ │ │ ├── apcnet_r101-d8_512x1024_80k_cityscapes.py │ │ │ ├── apcnet_r101-d8_512x512_160k_ade20k.py │ │ │ ├── apcnet_r101-d8_512x512_80k_ade20k.py │ │ │ ├── apcnet_r101-d8_769x769_40k_cityscapes.py │ │ │ ├── apcnet_r101-d8_769x769_80k_cityscapes.py │ │ │ ├── apcnet_r50-d8_512x1024_40k_cityscapes.py │ │ │ ├── apcnet_r50-d8_512x1024_80k_cityscapes.py │ │ │ ├── apcnet_r50-d8_512x512_160k_ade20k.py │ │ │ ├── apcnet_r50-d8_512x512_80k_ade20k.py │ │ │ ├── apcnet_r50-d8_769x769_40k_cityscapes.py │ │ │ └── apcnet_r50-d8_769x769_80k_cityscapes.py │ │ ├── bisenetv1 │ │ │ ├── README.md │ │ │ ├── bisenetv1.yml │ │ │ ├── bisenetv1_r101-d32_in1k-pre_lr5e-3_4x4_512x512_160k_coco-stuff164k.py │ │ │ ├── bisenetv1_r101-d32_lr5e-3_4x4_512x512_160k_coco-stuff164k.py │ │ │ ├── bisenetv1_r18-d32_4x4_1024x1024_160k_cityscapes.py │ │ │ ├── bisenetv1_r18-d32_in1k-pre_4x4_1024x1024_160k_cityscapes.py │ │ │ ├── bisenetv1_r18-d32_in1k-pre_4x8_1024x1024_160k_cityscapes.py │ │ │ ├── bisenetv1_r18-d32_in1k-pre_lr5e-3_4x4_512x512_160k_coco-stuff164k.py │ │ │ ├── bisenetv1_r18-d32_lr5e-3_4x4_512x512_160k_coco-stuff164k.py │ │ │ ├── bisenetv1_r50-d32_4x4_1024x1024_160k_cityscapes.py │ │ │ ├── bisenetv1_r50-d32_in1k-pre_4x4_1024x1024_160k_cityscapes.py │ │ │ ├── bisenetv1_r50-d32_in1k-pre_lr5e-3_4x4_512x512_160k_coco-stuff164k.py │ │ │ └── bisenetv1_r50-d32_lr5e-3_4x4_512x512_160k_coco-stuff164k.py │ │ ├── bisenetv2 │ │ │ ├── README.md │ │ │ ├── bisenetv2.yml │ │ │ ├── bisenetv2_fcn_4x4_1024x1024_160k_cityscapes.py │ │ │ ├── bisenetv2_fcn_4x8_1024x1024_160k_cityscapes.py │ │ │ ├── bisenetv2_fcn_fp16_4x4_1024x1024_160k_cityscapes.py │ │ │ └── bisenetv2_fcn_ohem_4x4_1024x1024_160k_cityscapes.py │ │ ├── ccnet │ │ │ ├── README.md │ │ │ ├── ccnet.yml │ │ │ ├── ccnet_r101-d8_512x1024_40k_cityscapes.py │ │ │ ├── ccnet_r101-d8_512x1024_80k_cityscapes.py │ │ │ ├── ccnet_r101-d8_512x512_160k_ade20k.py │ │ │ ├── ccnet_r101-d8_512x512_20k_voc12aug.py │ │ │ ├── ccnet_r101-d8_512x512_40k_voc12aug.py │ │ │ ├── ccnet_r101-d8_512x512_80k_ade20k.py │ │ │ ├── ccnet_r101-d8_769x769_40k_cityscapes.py │ │ │ ├── ccnet_r101-d8_769x769_80k_cityscapes.py │ │ │ ├── ccnet_r50-d8_512x1024_40k_cityscapes.py │ │ │ ├── ccnet_r50-d8_512x1024_80k_cityscapes.py │ │ │ ├── ccnet_r50-d8_512x512_160k_ade20k.py │ │ │ ├── ccnet_r50-d8_512x512_20k_voc12aug.py │ │ │ ├── ccnet_r50-d8_512x512_40k_voc12aug.py │ │ │ ├── ccnet_r50-d8_512x512_80k_ade20k.py │ │ │ ├── ccnet_r50-d8_769x769_40k_cityscapes.py │ │ │ └── ccnet_r50-d8_769x769_80k_cityscapes.py │ │ ├── cgnet │ │ │ ├── README.md │ │ │ ├── cgnet.yml │ │ │ ├── cgnet_512x1024_60k_cityscapes.py │ │ │ └── cgnet_680x680_60k_cityscapes.py │ │ ├── convnext │ │ │ ├── README.md │ │ │ ├── convnext.yml │ │ │ ├── upernet_convnext_base_fp16_512x512_160k_ade20k.py │ │ │ ├── upernet_convnext_base_fp16_640x640_160k_ade20k.py │ │ │ ├── upernet_convnext_large_fp16_640x640_160k_ade20k.py │ │ │ ├── upernet_convnext_small_fp16_512x512_160k_ade20k.py │ │ │ ├── upernet_convnext_tiny_fp16_512x512_160k_ade20k.py │ │ │ └── upernet_convnext_xlarge_fp16_640x640_160k_ade20k.py │ │ ├── danet │ │ │ ├── README.md │ │ │ ├── danet.yml │ │ │ ├── danet_r101-d8_512x1024_40k_cityscapes.py │ │ │ ├── danet_r101-d8_512x1024_80k_cityscapes.py │ │ │ ├── danet_r101-d8_512x512_160k_ade20k.py │ │ │ ├── danet_r101-d8_512x512_20k_voc12aug.py │ │ │ ├── danet_r101-d8_512x512_40k_voc12aug.py │ │ │ ├── danet_r101-d8_512x512_80k_ade20k.py │ │ │ ├── danet_r101-d8_769x769_40k_cityscapes.py │ │ │ ├── danet_r101-d8_769x769_80k_cityscapes.py │ │ │ ├── danet_r50-d8_512x1024_40k_cityscapes.py │ │ │ ├── danet_r50-d8_512x1024_80k_cityscapes.py │ │ │ ├── danet_r50-d8_512x512_160k_ade20k.py │ │ │ ├── danet_r50-d8_512x512_20k_voc12aug.py │ │ │ ├── danet_r50-d8_512x512_40k_voc12aug.py │ │ │ ├── danet_r50-d8_512x512_80k_ade20k.py │ │ │ ├── danet_r50-d8_769x769_40k_cityscapes.py │ │ │ └── danet_r50-d8_769x769_80k_cityscapes.py │ │ ├── deeplabv3 │ │ │ ├── README.md │ │ │ ├── deeplabv3.yml │ │ │ ├── deeplabv3_r101-d16-mg124_512x1024_40k_cityscapes.py │ │ │ ├── deeplabv3_r101-d16-mg124_512x1024_80k_cityscapes.py │ │ │ ├── deeplabv3_r101-d8_480x480_40k_pascal_context.py │ │ │ ├── deeplabv3_r101-d8_480x480_40k_pascal_context_59.py │ │ │ ├── deeplabv3_r101-d8_480x480_80k_pascal_context.py │ │ │ ├── deeplabv3_r101-d8_480x480_80k_pascal_context_59.py │ │ │ ├── deeplabv3_r101-d8_512x1024_40k_cityscapes.py │ │ │ ├── deeplabv3_r101-d8_512x1024_80k_cityscapes.py │ │ │ ├── deeplabv3_r101-d8_512x512_160k_ade20k.py │ │ │ ├── deeplabv3_r101-d8_512x512_20k_voc12aug.py │ │ │ ├── deeplabv3_r101-d8_512x512_40k_voc12aug.py │ │ │ ├── deeplabv3_r101-d8_512x512_4x4_160k_coco-stuff164k.py │ │ │ ├── deeplabv3_r101-d8_512x512_4x4_20k_coco-stuff10k.py │ │ │ ├── deeplabv3_r101-d8_512x512_4x4_320k_coco-stuff164k.py │ │ │ ├── deeplabv3_r101-d8_512x512_4x4_40k_coco-stuff10k.py │ │ │ ├── deeplabv3_r101-d8_512x512_4x4_80k_coco-stuff164k.py │ │ │ ├── deeplabv3_r101-d8_512x512_80k_ade20k.py │ │ │ ├── deeplabv3_r101-d8_769x769_40k_cityscapes.py │ │ │ ├── deeplabv3_r101-d8_769x769_80k_cityscapes.py │ │ │ ├── deeplabv3_r101-d8_fp16_512x1024_80k_cityscapes.py │ │ │ ├── deeplabv3_r101b-d8_512x1024_80k_cityscapes.py │ │ │ ├── deeplabv3_r101b-d8_769x769_80k_cityscapes.py │ │ │ ├── deeplabv3_r18-d8_512x1024_80k_cityscapes.py │ │ │ ├── deeplabv3_r18-d8_769x769_80k_cityscapes.py │ │ │ ├── deeplabv3_r18b-d8_512x1024_80k_cityscapes.py │ │ │ ├── deeplabv3_r18b-d8_769x769_80k_cityscapes.py │ │ │ ├── deeplabv3_r50-d8_480x480_40k_pascal_context.py │ │ │ ├── deeplabv3_r50-d8_480x480_40k_pascal_context_59.py │ │ │ ├── deeplabv3_r50-d8_480x480_80k_pascal_context.py │ │ │ ├── deeplabv3_r50-d8_480x480_80k_pascal_context_59.py │ │ │ ├── deeplabv3_r50-d8_512x1024_40k_cityscapes.py │ │ │ ├── deeplabv3_r50-d8_512x1024_80k_cityscapes.py │ │ │ ├── deeplabv3_r50-d8_512x512_160k_ade20k.py │ │ │ ├── deeplabv3_r50-d8_512x512_20k_voc12aug.py │ │ │ ├── deeplabv3_r50-d8_512x512_40k_voc12aug.py │ │ │ ├── deeplabv3_r50-d8_512x512_4x4_160k_coco-stuff164k.py │ │ │ ├── deeplabv3_r50-d8_512x512_4x4_20k_coco-stuff10k.py │ │ │ ├── deeplabv3_r50-d8_512x512_4x4_320k_coco-stuff164k.py │ │ │ ├── deeplabv3_r50-d8_512x512_4x4_40k_coco-stuff10k.py │ │ │ ├── deeplabv3_r50-d8_512x512_4x4_80k_coco-stuff164k.py │ │ │ ├── deeplabv3_r50-d8_512x512_80k_ade20k.py │ │ │ ├── deeplabv3_r50-d8_769x769_40k_cityscapes.py │ │ │ ├── deeplabv3_r50-d8_769x769_80k_cityscapes.py │ │ │ ├── deeplabv3_r50b-d8_512x1024_80k_cityscapes.py │ │ │ └── deeplabv3_r50b-d8_769x769_80k_cityscapes.py │ │ ├── deeplabv3plus │ │ │ ├── README.md │ │ │ ├── deeplabv3plus.yml │ │ │ ├── deeplabv3plus_r101-d16-mg124_512x1024_40k_cityscapes.py │ │ │ ├── deeplabv3plus_r101-d16-mg124_512x1024_80k_cityscapes.py │ │ │ ├── deeplabv3plus_r101-d8_480x480_40k_pascal_context.py │ │ │ ├── deeplabv3plus_r101-d8_480x480_40k_pascal_context_59.py │ │ │ ├── deeplabv3plus_r101-d8_480x480_80k_pascal_context.py │ │ │ ├── deeplabv3plus_r101-d8_480x480_80k_pascal_context_59.py │ │ │ ├── deeplabv3plus_r101-d8_4x4_512x512_80k_vaihingen.py │ │ │ ├── deeplabv3plus_r101-d8_512x1024_40k_cityscapes.py │ │ │ ├── deeplabv3plus_r101-d8_512x1024_80k_cityscapes.py │ │ │ ├── deeplabv3plus_r101-d8_512x512_160k_ade20k.py │ │ │ ├── deeplabv3plus_r101-d8_512x512_20k_voc12aug.py │ │ │ ├── deeplabv3plus_r101-d8_512x512_40k_voc12aug.py │ │ │ ├── deeplabv3plus_r101-d8_512x512_80k_ade20k.py │ │ │ ├── deeplabv3plus_r101-d8_512x512_80k_loveda.py │ │ │ ├── deeplabv3plus_r101-d8_512x512_80k_potsdam.py │ │ │ ├── deeplabv3plus_r101-d8_769x769_40k_cityscapes.py │ │ │ ├── deeplabv3plus_r101-d8_769x769_80k_cityscapes.py │ │ │ ├── deeplabv3plus_r101-d8_fp16_512x1024_80k_cityscapes.py │ │ │ ├── deeplabv3plus_r101b-d8_512x1024_80k_cityscapes.py │ │ │ ├── deeplabv3plus_r101b-d8_769x769_80k_cityscapes.py │ │ │ ├── deeplabv3plus_r18-d8_4x4_512x512_80k_vaihingen.py │ │ │ ├── deeplabv3plus_r18-d8_4x4_896x896_80k_isaid.py │ │ │ ├── deeplabv3plus_r18-d8_512x1024_80k_cityscapes.py │ │ │ ├── deeplabv3plus_r18-d8_512x512_80k_loveda.py │ │ │ ├── deeplabv3plus_r18-d8_512x512_80k_potsdam.py │ │ │ ├── deeplabv3plus_r18-d8_769x769_80k_cityscapes.py │ │ │ ├── deeplabv3plus_r18b-d8_512x1024_80k_cityscapes.py │ │ │ ├── deeplabv3plus_r18b-d8_769x769_80k_cityscapes.py │ │ │ ├── deeplabv3plus_r50-d8_480x480_40k_pascal_context.py │ │ │ ├── deeplabv3plus_r50-d8_480x480_40k_pascal_context_59.py │ │ │ ├── deeplabv3plus_r50-d8_480x480_80k_pascal_context.py │ │ │ ├── deeplabv3plus_r50-d8_480x480_80k_pascal_context_59.py │ │ │ ├── deeplabv3plus_r50-d8_4x4_512x512_80k_vaihingen.py │ │ │ ├── deeplabv3plus_r50-d8_4x4_896x896_80k_isaid.py │ │ │ ├── deeplabv3plus_r50-d8_512x1024_40k_cityscapes.py │ │ │ ├── deeplabv3plus_r50-d8_512x1024_80k_cityscapes.py │ │ │ ├── deeplabv3plus_r50-d8_512x512_160k_ade20k.py │ │ │ ├── deeplabv3plus_r50-d8_512x512_20k_voc12aug.py │ │ │ ├── deeplabv3plus_r50-d8_512x512_40k_voc12aug.py │ │ │ ├── deeplabv3plus_r50-d8_512x512_80k_ade20k.py │ │ │ ├── deeplabv3plus_r50-d8_512x512_80k_loveda.py │ │ │ ├── deeplabv3plus_r50-d8_512x512_80k_potsdam.py │ │ │ ├── deeplabv3plus_r50-d8_769x769_40k_cityscapes.py │ │ │ ├── deeplabv3plus_r50-d8_769x769_80k_cityscapes.py │ │ │ ├── deeplabv3plus_r50b-d8_512x1024_80k_cityscapes.py │ │ │ └── deeplabv3plus_r50b-d8_769x769_80k_cityscapes.py │ │ ├── dmnet │ │ │ ├── README.md │ │ │ ├── dmnet.yml │ │ │ ├── dmnet_r101-d8_512x1024_40k_cityscapes.py │ │ │ ├── dmnet_r101-d8_512x1024_80k_cityscapes.py │ │ │ ├── dmnet_r101-d8_512x512_160k_ade20k.py │ │ │ ├── dmnet_r101-d8_512x512_80k_ade20k.py │ │ │ ├── dmnet_r101-d8_769x769_40k_cityscapes.py │ │ │ ├── dmnet_r101-d8_769x769_80k_cityscapes.py │ │ │ ├── dmnet_r50-d8_512x1024_40k_cityscapes.py │ │ │ ├── dmnet_r50-d8_512x1024_80k_cityscapes.py │ │ │ ├── dmnet_r50-d8_512x512_160k_ade20k.py │ │ │ ├── dmnet_r50-d8_512x512_80k_ade20k.py │ │ │ ├── dmnet_r50-d8_769x769_40k_cityscapes.py │ │ │ └── dmnet_r50-d8_769x769_80k_cityscapes.py │ │ ├── dnlnet │ │ │ ├── README.md │ │ │ ├── dnl_r101-d8_512x1024_40k_cityscapes.py │ │ │ ├── dnl_r101-d8_512x1024_80k_cityscapes.py │ │ │ ├── dnl_r101-d8_512x512_160k_ade20k.py │ │ │ ├── dnl_r101-d8_512x512_80k_ade20k.py │ │ │ ├── dnl_r101-d8_769x769_40k_cityscapes.py │ │ │ ├── dnl_r101-d8_769x769_80k_cityscapes.py │ │ │ ├── dnl_r50-d8_512x1024_40k_cityscapes.py │ │ │ ├── dnl_r50-d8_512x1024_80k_cityscapes.py │ │ │ ├── dnl_r50-d8_512x512_160k_ade20k.py │ │ │ ├── dnl_r50-d8_512x512_80k_ade20k.py │ │ │ ├── dnl_r50-d8_769x769_40k_cityscapes.py │ │ │ ├── dnl_r50-d8_769x769_80k_cityscapes.py │ │ │ └── dnlnet.yml │ │ ├── dpt │ │ │ ├── README.md │ │ │ ├── dpt.yml │ │ │ └── dpt_vit-b16_512x512_160k_ade20k.py │ │ ├── emanet │ │ │ ├── README.md │ │ │ ├── emanet.yml │ │ │ ├── emanet_r101-d8_512x1024_80k_cityscapes.py │ │ │ ├── emanet_r101-d8_769x769_80k_cityscapes.py │ │ │ ├── emanet_r50-d8_512x1024_80k_cityscapes.py │ │ │ └── emanet_r50-d8_769x769_80k_cityscapes.py │ │ ├── encnet │ │ │ ├── README.md │ │ │ ├── encnet.yml │ │ │ ├── encnet_r101-d8_512x1024_40k_cityscapes.py │ │ │ ├── encnet_r101-d8_512x1024_80k_cityscapes.py │ │ │ ├── encnet_r101-d8_512x512_160k_ade20k.py │ │ │ ├── encnet_r101-d8_512x512_20k_voc12aug.py │ │ │ ├── encnet_r101-d8_512x512_40k_voc12aug.py │ │ │ ├── encnet_r101-d8_512x512_80k_ade20k.py │ │ │ ├── encnet_r101-d8_769x769_40k_cityscapes.py │ │ │ ├── encnet_r101-d8_769x769_80k_cityscapes.py │ │ │ ├── encnet_r50-d8_512x1024_40k_cityscapes.py │ │ │ ├── encnet_r50-d8_512x1024_80k_cityscapes.py │ │ │ ├── encnet_r50-d8_512x512_160k_ade20k.py │ │ │ ├── encnet_r50-d8_512x512_20k_voc12aug.py │ │ │ ├── encnet_r50-d8_512x512_40k_voc12aug.py │ │ │ ├── encnet_r50-d8_512x512_80k_ade20k.py │ │ │ ├── encnet_r50-d8_769x769_40k_cityscapes.py │ │ │ ├── encnet_r50-d8_769x769_80k_cityscapes.py │ │ │ └── encnet_r50s-d8_512x512_80k_ade20k.py │ │ ├── erfnet │ │ │ ├── README.md │ │ │ ├── erfnet.yml │ │ │ └── erfnet_fcn_4x4_512x1024_160k_cityscapes.py │ │ ├── fastfcn │ │ │ ├── README.md │ │ │ ├── fastfcn.yml │ │ │ ├── fastfcn_r50-d32_jpu_aspp_4x4_512x1024_80k_cityscapes.py │ │ │ ├── fastfcn_r50-d32_jpu_aspp_512x1024_80k_cityscapes.py │ │ │ ├── fastfcn_r50-d32_jpu_aspp_512x512_160k_ade20k.py │ │ │ ├── fastfcn_r50-d32_jpu_aspp_512x512_80k_ade20k.py │ │ │ ├── fastfcn_r50-d32_jpu_enc_4x4_512x1024_80k_cityscapes.py │ │ │ ├── fastfcn_r50-d32_jpu_enc_512x1024_80k_cityscapes.py │ │ │ ├── fastfcn_r50-d32_jpu_enc_512x512_160k_ade20k.py │ │ │ ├── fastfcn_r50-d32_jpu_enc_512x512_80k_ade20k.py │ │ │ ├── fastfcn_r50-d32_jpu_psp_4x4_512x1024_80k_cityscapes.py │ │ │ ├── fastfcn_r50-d32_jpu_psp_512x1024_80k_cityscapes.py │ │ │ ├── fastfcn_r50-d32_jpu_psp_512x512_160k_ade20k.py │ │ │ └── fastfcn_r50-d32_jpu_psp_512x512_80k_ade20k.py │ │ ├── fastscnn │ │ │ ├── README.md │ │ │ ├── fast_scnn_lr0.12_8x4_160k_cityscapes.py │ │ │ └── fastscnn.yml │ │ ├── fcn │ │ │ ├── README.md │ │ │ ├── fcn.yml │ │ │ ├── fcn_d6_r101-d16_512x1024_40k_cityscapes.py │ │ │ ├── fcn_d6_r101-d16_512x1024_80k_cityscapes.py │ │ │ ├── fcn_d6_r101-d16_769x769_40k_cityscapes.py │ │ │ ├── fcn_d6_r101-d16_769x769_80k_cityscapes.py │ │ │ ├── fcn_d6_r101b-d16_512x1024_80k_cityscapes.py │ │ │ ├── fcn_d6_r101b-d16_769x769_80k_cityscapes.py │ │ │ ├── fcn_d6_r50-d16_512x1024_40k_cityscapes.py │ │ │ ├── fcn_d6_r50-d16_512x1024_80k_cityscapes.py │ │ │ ├── fcn_d6_r50-d16_769x769_40k_cityscapes.py │ │ │ ├── fcn_d6_r50-d16_769x769_80k_cityscapes.py │ │ │ ├── fcn_d6_r50b-d16_512x1024_80k_cityscapes.py │ │ │ ├── fcn_d6_r50b-d16_769x769_80k_cityscapes.py │ │ │ ├── fcn_r101-d8_480x480_40k_pascal_context.py │ │ │ ├── fcn_r101-d8_480x480_40k_pascal_context_59.py │ │ │ ├── fcn_r101-d8_480x480_80k_pascal_context.py │ │ │ ├── fcn_r101-d8_480x480_80k_pascal_context_59.py │ │ │ ├── fcn_r101-d8_512x1024_40k_cityscapes.py │ │ │ ├── fcn_r101-d8_512x1024_80k_cityscapes.py │ │ │ ├── fcn_r101-d8_512x512_160k_ade20k.py │ │ │ ├── fcn_r101-d8_512x512_20k_voc12aug.py │ │ │ ├── fcn_r101-d8_512x512_40k_voc12aug.py │ │ │ ├── fcn_r101-d8_512x512_80k_ade20k.py │ │ │ ├── fcn_r101-d8_769x769_40k_cityscapes.py │ │ │ ├── fcn_r101-d8_769x769_80k_cityscapes.py │ │ │ ├── fcn_r101-d8_fp16_512x1024_80k_cityscapes.py │ │ │ ├── fcn_r101b-d8_512x1024_80k_cityscapes.py │ │ │ ├── fcn_r101b-d8_769x769_80k_cityscapes.py │ │ │ ├── fcn_r18-d8_512x1024_80k_cityscapes.py │ │ │ ├── fcn_r18-d8_769x769_80k_cityscapes.py │ │ │ ├── fcn_r18b-d8_512x1024_80k_cityscapes.py │ │ │ ├── fcn_r18b-d8_769x769_80k_cityscapes.py │ │ │ ├── fcn_r50-d8_480x480_40k_pascal_context.py │ │ │ ├── fcn_r50-d8_480x480_40k_pascal_context_59.py │ │ │ ├── fcn_r50-d8_480x480_80k_pascal_context.py │ │ │ ├── fcn_r50-d8_480x480_80k_pascal_context_59.py │ │ │ ├── fcn_r50-d8_512x1024_40k_cityscapes.py │ │ │ ├── fcn_r50-d8_512x1024_80k_cityscapes.py │ │ │ ├── fcn_r50-d8_512x512_160k_ade20k.py │ │ │ ├── fcn_r50-d8_512x512_20k_voc12aug.py │ │ │ ├── fcn_r50-d8_512x512_40k_voc12aug.py │ │ │ ├── fcn_r50-d8_512x512_80k_ade20k.py │ │ │ ├── fcn_r50-d8_769x769_40k_cityscapes.py │ │ │ ├── fcn_r50-d8_769x769_80k_cityscapes.py │ │ │ ├── fcn_r50b-d8_512x1024_80k_cityscapes.py │ │ │ └── fcn_r50b-d8_769x769_80k_cityscapes.py │ │ ├── gcnet │ │ │ ├── README.md │ │ │ ├── gcnet.yml │ │ │ ├── gcnet_r101-d8_512x1024_40k_cityscapes.py │ │ │ ├── gcnet_r101-d8_512x1024_80k_cityscapes.py │ │ │ ├── gcnet_r101-d8_512x512_160k_ade20k.py │ │ │ ├── gcnet_r101-d8_512x512_20k_voc12aug.py │ │ │ ├── gcnet_r101-d8_512x512_40k_voc12aug.py │ │ │ ├── gcnet_r101-d8_512x512_80k_ade20k.py │ │ │ ├── gcnet_r101-d8_769x769_40k_cityscapes.py │ │ │ ├── gcnet_r101-d8_769x769_80k_cityscapes.py │ │ │ ├── gcnet_r50-d8_512x1024_40k_cityscapes.py │ │ │ ├── gcnet_r50-d8_512x1024_80k_cityscapes.py │ │ │ ├── gcnet_r50-d8_512x512_160k_ade20k.py │ │ │ ├── gcnet_r50-d8_512x512_20k_voc12aug.py │ │ │ ├── gcnet_r50-d8_512x512_40k_voc12aug.py │ │ │ ├── gcnet_r50-d8_512x512_80k_ade20k.py │ │ │ ├── gcnet_r50-d8_769x769_40k_cityscapes.py │ │ │ └── gcnet_r50-d8_769x769_80k_cityscapes.py │ │ ├── hrnet │ │ │ ├── README.md │ │ │ ├── fcn_hr18_480x480_40k_pascal_context.py │ │ │ ├── fcn_hr18_480x480_40k_pascal_context_59.py │ │ │ ├── fcn_hr18_480x480_80k_pascal_context.py │ │ │ ├── fcn_hr18_480x480_80k_pascal_context_59.py │ │ │ ├── fcn_hr18_4x4_512x512_80k_vaihingen.py │ │ │ ├── fcn_hr18_4x4_896x896_80k_isaid.py │ │ │ ├── fcn_hr18_512x1024_160k_cityscapes.py │ │ │ ├── fcn_hr18_512x1024_40k_cityscapes.py │ │ │ ├── fcn_hr18_512x1024_80k_cityscapes.py │ │ │ ├── fcn_hr18_512x512_160k_ade20k.py │ │ │ ├── fcn_hr18_512x512_20k_voc12aug.py │ │ │ ├── fcn_hr18_512x512_40k_voc12aug.py │ │ │ ├── fcn_hr18_512x512_80k_ade20k.py │ │ │ ├── fcn_hr18_512x512_80k_loveda.py │ │ │ ├── fcn_hr18_512x512_80k_potsdam.py │ │ │ ├── fcn_hr18s_480x480_40k_pascal_context.py │ │ │ ├── fcn_hr18s_480x480_40k_pascal_context_59.py │ │ │ ├── fcn_hr18s_480x480_80k_pascal_context.py │ │ │ ├── fcn_hr18s_480x480_80k_pascal_context_59.py │ │ │ ├── fcn_hr18s_4x4_512x512_80k_vaihingen.py │ │ │ ├── fcn_hr18s_4x4_896x896_80k_isaid.py │ │ │ ├── fcn_hr18s_512x1024_160k_cityscapes.py │ │ │ ├── fcn_hr18s_512x1024_40k_cityscapes.py │ │ │ ├── fcn_hr18s_512x1024_80k_cityscapes.py │ │ │ ├── fcn_hr18s_512x512_160k_ade20k.py │ │ │ ├── fcn_hr18s_512x512_20k_voc12aug.py │ │ │ ├── fcn_hr18s_512x512_40k_voc12aug.py │ │ │ ├── fcn_hr18s_512x512_80k_ade20k.py │ │ │ ├── fcn_hr18s_512x512_80k_loveda.py │ │ │ ├── fcn_hr18s_512x512_80k_potsdam.py │ │ │ ├── fcn_hr48_480x480_40k_pascal_context.py │ │ │ ├── fcn_hr48_480x480_40k_pascal_context_59.py │ │ │ ├── fcn_hr48_480x480_80k_pascal_context.py │ │ │ ├── fcn_hr48_480x480_80k_pascal_context_59.py │ │ │ ├── fcn_hr48_4x4_512x512_80k_vaihingen.py │ │ │ ├── fcn_hr48_4x4_896x896_80k_isaid.py │ │ │ ├── fcn_hr48_512x1024_160k_cityscapes.py │ │ │ ├── fcn_hr48_512x1024_40k_cityscapes.py │ │ │ ├── fcn_hr48_512x1024_80k_cityscapes.py │ │ │ ├── fcn_hr48_512x512_160k_ade20k.py │ │ │ ├── fcn_hr48_512x512_20k_voc12aug.py │ │ │ ├── fcn_hr48_512x512_40k_voc12aug.py │ │ │ ├── fcn_hr48_512x512_80k_ade20k.py │ │ │ ├── fcn_hr48_512x512_80k_loveda.py │ │ │ ├── fcn_hr48_512x512_80k_potsdam.py │ │ │ └── hrnet.yml │ │ ├── icnet │ │ │ ├── README.md │ │ │ ├── icnet.yml │ │ │ ├── icnet_r101-d8_832x832_160k_cityscapes.py │ │ │ ├── icnet_r101-d8_832x832_80k_cityscapes.py │ │ │ ├── icnet_r101-d8_in1k-pre_832x832_160k_cityscapes.py │ │ │ ├── icnet_r101-d8_in1k-pre_832x832_80k_cityscapes.py │ │ │ ├── icnet_r18-d8_832x832_160k_cityscapes.py │ │ │ ├── icnet_r18-d8_832x832_80k_cityscapes.py │ │ │ ├── icnet_r18-d8_in1k-pre_832x832_160k_cityscapes.py │ │ │ ├── icnet_r18-d8_in1k-pre_832x832_80k_cityscapes.py │ │ │ ├── icnet_r50-d8_832x832_160k_cityscapes.py │ │ │ ├── icnet_r50-d8_832x832_80k_cityscapes.py │ │ │ ├── icnet_r50-d8_in1k-pre_832x832_160k_cityscapes.py │ │ │ └── icnet_r50-d8_in1k-pre_832x832_80k_cityscapes.py │ │ ├── isanet │ │ │ ├── README.md │ │ │ ├── isanet.yml │ │ │ ├── isanet_r101-d8_512x1024_40k_cityscapes.py │ │ │ ├── isanet_r101-d8_512x1024_80k_cityscapes.py │ │ │ ├── isanet_r101-d8_512x512_160k_ade20k.py │ │ │ ├── isanet_r101-d8_512x512_20k_voc12aug.py │ │ │ ├── isanet_r101-d8_512x512_40k_voc12aug.py │ │ │ ├── isanet_r101-d8_512x512_80k_ade20k.py │ │ │ ├── isanet_r101-d8_769x769_40k_cityscapes.py │ │ │ ├── isanet_r101-d8_769x769_80k_cityscapes.py │ │ │ ├── isanet_r50-d8_512x1024_40k_cityscapes.py │ │ │ ├── isanet_r50-d8_512x1024_80k_cityscapes.py │ │ │ ├── isanet_r50-d8_512x512_160k_ade20k.py │ │ │ ├── isanet_r50-d8_512x512_20k_voc12aug.py │ │ │ ├── isanet_r50-d8_512x512_40k_voc12aug.py │ │ │ ├── isanet_r50-d8_512x512_80k_ade20k.py │ │ │ ├── isanet_r50-d8_769x769_40k_cityscapes.py │ │ │ └── isanet_r50-d8_769x769_80k_cityscapes.py │ │ ├── mobilenet_v2 │ │ │ ├── README.md │ │ │ ├── deeplabv3_m-v2-d8_512x1024_80k_cityscapes.py │ │ │ ├── deeplabv3_m-v2-d8_512x512_160k_ade20k.py │ │ │ ├── deeplabv3plus_m-v2-d8_512x1024_80k_cityscapes.py │ │ │ ├── deeplabv3plus_m-v2-d8_512x512_160k_ade20k.py │ │ │ ├── fcn_m-v2-d8_512x1024_80k_cityscapes.py │ │ │ ├── fcn_m-v2-d8_512x512_160k_ade20k.py │ │ │ ├── mobilenet_v2.yml │ │ │ ├── pspnet_m-v2-d8_512x1024_80k_cityscapes.py │ │ │ └── pspnet_m-v2-d8_512x512_160k_ade20k.py │ │ ├── mobilenet_v3 │ │ │ ├── README.md │ │ │ ├── lraspp_m-v3-d8_512x1024_320k_cityscapes.py │ │ │ ├── lraspp_m-v3-d8_scratch_512x1024_320k_cityscapes.py │ │ │ ├── lraspp_m-v3s-d8_512x1024_320k_cityscapes.py │ │ │ ├── lraspp_m-v3s-d8_scratch_512x1024_320k_cityscapes.py │ │ │ └── mobilenet_v3.yml │ │ ├── nonlocal_net │ │ │ ├── README.md │ │ │ ├── nonlocal_net.yml │ │ │ ├── nonlocal_r101-d8_512x1024_40k_cityscapes.py │ │ │ ├── nonlocal_r101-d8_512x1024_80k_cityscapes.py │ │ │ ├── nonlocal_r101-d8_512x512_160k_ade20k.py │ │ │ ├── nonlocal_r101-d8_512x512_20k_voc12aug.py │ │ │ ├── nonlocal_r101-d8_512x512_40k_voc12aug.py │ │ │ ├── nonlocal_r101-d8_512x512_80k_ade20k.py │ │ │ ├── nonlocal_r101-d8_769x769_40k_cityscapes.py │ │ │ ├── nonlocal_r101-d8_769x769_80k_cityscapes.py │ │ │ ├── nonlocal_r50-d8_512x1024_40k_cityscapes.py │ │ │ ├── nonlocal_r50-d8_512x1024_80k_cityscapes.py │ │ │ ├── nonlocal_r50-d8_512x512_160k_ade20k.py │ │ │ ├── nonlocal_r50-d8_512x512_20k_voc12aug.py │ │ │ ├── nonlocal_r50-d8_512x512_40k_voc12aug.py │ │ │ ├── nonlocal_r50-d8_512x512_80k_ade20k.py │ │ │ ├── nonlocal_r50-d8_769x769_40k_cityscapes.py │ │ │ └── nonlocal_r50-d8_769x769_80k_cityscapes.py │ │ ├── ocrnet │ │ │ ├── README.md │ │ │ ├── ocrnet.yml │ │ │ ├── ocrnet_hr18_512x1024_160k_cityscapes.py │ │ │ ├── ocrnet_hr18_512x1024_40k_cityscapes.py │ │ │ ├── ocrnet_hr18_512x1024_80k_cityscapes.py │ │ │ ├── ocrnet_hr18_512x512_160k_ade20k.py │ │ │ ├── ocrnet_hr18_512x512_20k_voc12aug.py │ │ │ ├── ocrnet_hr18_512x512_40k_voc12aug.py │ │ │ ├── ocrnet_hr18_512x512_80k_ade20k.py │ │ │ ├── ocrnet_hr18s_512x1024_160k_cityscapes.py │ │ │ ├── ocrnet_hr18s_512x1024_40k_cityscapes.py │ │ │ ├── ocrnet_hr18s_512x1024_80k_cityscapes.py │ │ │ ├── ocrnet_hr18s_512x512_160k_ade20k.py │ │ │ ├── ocrnet_hr18s_512x512_20k_voc12aug.py │ │ │ ├── ocrnet_hr18s_512x512_40k_voc12aug.py │ │ │ ├── ocrnet_hr18s_512x512_80k_ade20k.py │ │ │ ├── ocrnet_hr48_512x1024_160k_cityscapes.py │ │ │ ├── ocrnet_hr48_512x1024_40k_cityscapes.py │ │ │ ├── ocrnet_hr48_512x1024_80k_cityscapes.py │ │ │ ├── ocrnet_hr48_512x512_160k_ade20k.py │ │ │ ├── ocrnet_hr48_512x512_20k_voc12aug.py │ │ │ ├── ocrnet_hr48_512x512_40k_voc12aug.py │ │ │ ├── ocrnet_hr48_512x512_80k_ade20k.py │ │ │ ├── ocrnet_r101-d8_512x1024_40k_b16_cityscapes.py │ │ │ ├── ocrnet_r101-d8_512x1024_40k_b8_cityscapes.py │ │ │ └── ocrnet_r101-d8_512x1024_80k_b16_cityscapes.py │ │ ├── point_rend │ │ │ ├── README.md │ │ │ ├── point_rend.yml │ │ │ ├── pointrend_r101_512x1024_80k_cityscapes.py │ │ │ ├── pointrend_r101_512x512_160k_ade20k.py │ │ │ ├── pointrend_r50_512x1024_80k_cityscapes.py │ │ │ └── pointrend_r50_512x512_160k_ade20k.py │ │ ├── psanet │ │ │ ├── README.md │ │ │ ├── psanet.yml │ │ │ ├── psanet_r101-d8_512x1024_40k_cityscapes.py │ │ │ ├── psanet_r101-d8_512x1024_80k_cityscapes.py │ │ │ ├── psanet_r101-d8_512x512_160k_ade20k.py │ │ │ ├── psanet_r101-d8_512x512_20k_voc12aug.py │ │ │ ├── psanet_r101-d8_512x512_40k_voc12aug.py │ │ │ ├── psanet_r101-d8_512x512_80k_ade20k.py │ │ │ ├── psanet_r101-d8_769x769_40k_cityscapes.py │ │ │ ├── psanet_r101-d8_769x769_80k_cityscapes.py │ │ │ ├── psanet_r50-d8_512x1024_40k_cityscapes.py │ │ │ ├── psanet_r50-d8_512x1024_80k_cityscapes.py │ │ │ ├── psanet_r50-d8_512x512_160k_ade20k.py │ │ │ ├── psanet_r50-d8_512x512_20k_voc12aug.py │ │ │ ├── psanet_r50-d8_512x512_40k_voc12aug.py │ │ │ ├── psanet_r50-d8_512x512_80k_ade20k.py │ │ │ ├── psanet_r50-d8_769x769_40k_cityscapes.py │ │ │ └── psanet_r50-d8_769x769_80k_cityscapes.py │ │ ├── pspnet │ │ │ ├── README.md │ │ │ ├── pspnet.yml │ │ │ ├── pspnet_r101-d8_480x480_40k_pascal_context.py │ │ │ ├── pspnet_r101-d8_480x480_40k_pascal_context_59.py │ │ │ ├── pspnet_r101-d8_480x480_80k_pascal_context.py │ │ │ ├── pspnet_r101-d8_480x480_80k_pascal_context_59.py │ │ │ ├── pspnet_r101-d8_4x4_512x512_80k_potsdam.py │ │ │ ├── pspnet_r101-d8_4x4_512x512_80k_vaihingen.py │ │ │ ├── pspnet_r101-d8_512x1024_40k_cityscapes.py │ │ │ ├── pspnet_r101-d8_512x1024_40k_dark.py │ │ │ ├── pspnet_r101-d8_512x1024_40k_night_driving.py │ │ │ ├── pspnet_r101-d8_512x1024_80k_cityscapes.py │ │ │ ├── pspnet_r101-d8_512x512_160k_ade20k.py │ │ │ ├── pspnet_r101-d8_512x512_20k_voc12aug.py │ │ │ ├── pspnet_r101-d8_512x512_40k_voc12aug.py │ │ │ ├── pspnet_r101-d8_512x512_4x4_160k_coco-stuff164k.py │ │ │ ├── pspnet_r101-d8_512x512_4x4_20k_coco-stuff10k.py │ │ │ ├── pspnet_r101-d8_512x512_4x4_320k_coco-stuff164k.py │ │ │ ├── pspnet_r101-d8_512x512_4x4_40k_coco-stuff10k.py │ │ │ ├── pspnet_r101-d8_512x512_4x4_80k_coco-stuff164k.py │ │ │ ├── pspnet_r101-d8_512x512_80k_ade20k.py │ │ │ ├── pspnet_r101-d8_512x512_80k_loveda.py │ │ │ ├── pspnet_r101-d8_769x769_40k_cityscapes.py │ │ │ ├── pspnet_r101-d8_769x769_80k_cityscapes.py │ │ │ ├── pspnet_r101-d8_fp16_512x1024_80k_cityscapes.py │ │ │ ├── pspnet_r101b-d8_512x1024_80k_cityscapes.py │ │ │ ├── pspnet_r101b-d8_512x1024_80k_dark.py │ │ │ ├── pspnet_r101b-d8_512x1024_80k_night_driving.py │ │ │ ├── pspnet_r101b-d8_769x769_80k_cityscapes.py │ │ │ ├── pspnet_r18-d8_4x4_512x512_80k_potsdam.py │ │ │ ├── pspnet_r18-d8_4x4_512x512_80k_vaihingen.py │ │ │ ├── pspnet_r18-d8_4x4_896x896_80k_isaid.py │ │ │ ├── pspnet_r18-d8_512x1024_80k_cityscapes.py │ │ │ ├── pspnet_r18-d8_512x512_80k_loveda.py │ │ │ ├── pspnet_r18-d8_769x769_80k_cityscapes.py │ │ │ ├── pspnet_r18b-d8_512x1024_80k_cityscapes.py │ │ │ ├── pspnet_r18b-d8_769x769_80k_cityscapes.py │ │ │ ├── pspnet_r50-d8_480x480_40k_pascal_context.py │ │ │ ├── pspnet_r50-d8_480x480_40k_pascal_context_59.py │ │ │ ├── pspnet_r50-d8_480x480_80k_pascal_context.py │ │ │ ├── pspnet_r50-d8_480x480_80k_pascal_context_59.py │ │ │ ├── pspnet_r50-d8_4x4_512x512_80k_potsdam.py │ │ │ ├── pspnet_r50-d8_4x4_512x512_80k_vaihingen.py │ │ │ ├── pspnet_r50-d8_4x4_896x896_80k_isaid.py │ │ │ ├── pspnet_r50-d8_512x1024_40k_cityscapes.py │ │ │ ├── pspnet_r50-d8_512x1024_40k_dark.py │ │ │ ├── pspnet_r50-d8_512x1024_40k_night_driving.py │ │ │ ├── pspnet_r50-d8_512x1024_80k_cityscapes.py │ │ │ ├── pspnet_r50-d8_512x1024_80k_dark.py │ │ │ ├── pspnet_r50-d8_512x1024_80k_night_driving.py │ │ │ ├── pspnet_r50-d8_512x512_160k_ade20k.py │ │ │ ├── pspnet_r50-d8_512x512_20k_voc12aug.py │ │ │ ├── pspnet_r50-d8_512x512_40k_voc12aug.py │ │ │ ├── pspnet_r50-d8_512x512_4x4_160k_coco-stuff164k.py │ │ │ ├── pspnet_r50-d8_512x512_4x4_20k_coco-stuff10k.py │ │ │ ├── pspnet_r50-d8_512x512_4x4_320k_coco-stuff164k.py │ │ │ ├── pspnet_r50-d8_512x512_4x4_40k_coco-stuff10k.py │ │ │ ├── pspnet_r50-d8_512x512_4x4_80k_coco-stuff164k.py │ │ │ ├── pspnet_r50-d8_512x512_80k_ade20k.py │ │ │ ├── pspnet_r50-d8_512x512_80k_loveda.py │ │ │ ├── pspnet_r50-d8_769x769_40k_cityscapes.py │ │ │ ├── pspnet_r50-d8_769x769_80k_cityscapes.py │ │ │ ├── pspnet_r50b-d8_512x1024_80k_cityscapes.py │ │ │ └── pspnet_r50b-d8_769x769_80k_cityscapes.py │ │ ├── resnest │ │ │ ├── README.md │ │ │ ├── deeplabv3_s101-d8_512x1024_80k_cityscapes.py │ │ │ ├── deeplabv3_s101-d8_512x512_160k_ade20k.py │ │ │ ├── deeplabv3plus_s101-d8_512x1024_80k_cityscapes.py │ │ │ ├── deeplabv3plus_s101-d8_512x512_160k_ade20k.py │ │ │ ├── fcn_s101-d8_512x1024_80k_cityscapes.py │ │ │ ├── fcn_s101-d8_512x512_160k_ade20k.py │ │ │ ├── pspnet_s101-d8_512x1024_80k_cityscapes.py │ │ │ ├── pspnet_s101-d8_512x512_160k_ade20k.py │ │ │ └── resnest.yml │ │ ├── segformer │ │ │ ├── README.md │ │ │ ├── segformer.yml │ │ │ ├── segformer_mit-b0_512x512_160k_ade20k.py │ │ │ ├── segformer_mit-b0_8x1_1024x1024_160k_cityscapes.py │ │ │ ├── segformer_mit-b1_512x512_160k_ade20k.py │ │ │ ├── segformer_mit-b1_8x1_1024x1024_160k_cityscapes.py │ │ │ ├── segformer_mit-b2_512x512_160k_ade20k.py │ │ │ ├── segformer_mit-b2_8x1_1024x1024_160k_cityscapes.py │ │ │ ├── segformer_mit-b3_512x512_160k_ade20k.py │ │ │ ├── segformer_mit-b3_8x1_1024x1024_160k_cityscapes.py │ │ │ ├── segformer_mit-b4_512x512_160k_ade20k.py │ │ │ ├── segformer_mit-b4_8x1_1024x1024_160k_cityscapes.py │ │ │ ├── segformer_mit-b5_512x512_160k_ade20k.py │ │ │ ├── segformer_mit-b5_640x640_160k_ade20k.py │ │ │ └── segformer_mit-b5_8x1_1024x1024_160k_cityscapes.py │ │ ├── segmenter │ │ │ ├── README.md │ │ │ ├── segmenter.yml │ │ │ ├── segmenter_vit-b_mask_8x1_512x512_160k_ade20k.py │ │ │ ├── segmenter_vit-l_mask_8x1_512x512_160k_ade20k.py │ │ │ ├── segmenter_vit-s_linear_8x1_512x512_160k_ade20k.py │ │ │ ├── segmenter_vit-s_mask_8x1_512x512_160k_ade20k.py │ │ │ └── segmenter_vit-t_mask_8x1_512x512_160k_ade20k.py │ │ ├── sem_fpn │ │ │ ├── README.md │ │ │ ├── fpn_r101_512x1024_80k_cityscapes.py │ │ │ ├── fpn_r101_512x512_160k_ade20k.py │ │ │ ├── fpn_r50_512x1024_80k_cityscapes.py │ │ │ ├── fpn_r50_512x512_160k_ade20k.py │ │ │ └── sem_fpn.yml │ │ ├── setr │ │ │ ├── README.md │ │ │ ├── setr.yml │ │ │ ├── setr_mla_512x512_160k_b16_ade20k.py │ │ │ ├── setr_mla_512x512_160k_b8_ade20k.py │ │ │ ├── setr_naive_512x512_160k_b16_ade20k.py │ │ │ ├── setr_pup_512x512_160k_b16_ade20k.py │ │ │ ├── setr_vit-large_mla_8x1_768x768_80k_cityscapes.py │ │ │ ├── setr_vit-large_naive_8x1_768x768_80k_cityscapes.py │ │ │ └── setr_vit-large_pup_8x1_768x768_80k_cityscapes.py │ │ ├── stdc │ │ │ ├── README.md │ │ │ ├── stdc.yml │ │ │ ├── stdc1_512x1024_80k_cityscapes.py │ │ │ ├── stdc1_in1k-pre_512x1024_80k_cityscapes.py │ │ │ ├── stdc2_512x1024_80k_cityscapes.py │ │ │ └── stdc2_in1k-pre_512x1024_80k_cityscapes.py │ │ ├── swin │ │ │ ├── README.md │ │ │ ├── swin.yml │ │ │ ├── upernet_swin_base_patch4_window12_512x512_160k_ade20k_pretrain_384x384_1K.py │ │ │ ├── upernet_swin_base_patch4_window12_512x512_160k_ade20k_pretrain_384x384_22K.py │ │ │ ├── upernet_swin_base_patch4_window7_512x512_160k_ade20k_pretrain_224x224_1K.py │ │ │ ├── upernet_swin_base_patch4_window7_512x512_160k_ade20k_pretrain_224x224_22K.py │ │ │ ├── upernet_swin_small_patch4_window7_512x512_160k_ade20k_pretrain_224x224_1K.py │ │ │ └── upernet_swin_tiny_patch4_window7_512x512_160k_ade20k_pretrain_224x224_1K.py │ │ ├── twins │ │ │ ├── README.md │ │ │ ├── twins.yml │ │ │ ├── twins_pcpvt-b_fpn_fpnhead_8x4_512x512_80k_ade20k.py │ │ │ ├── twins_pcpvt-b_uperhead_8x2_512x512_160k_ade20k.py │ │ │ ├── twins_pcpvt-l_fpn_fpnhead_8x4_512x512_80k_ade20k.py │ │ │ ├── twins_pcpvt-l_uperhead_8x2_512x512_160k_ade20k.py │ │ │ ├── twins_pcpvt-s_fpn_fpnhead_8x4_512x512_80k_ade20k.py │ │ │ ├── twins_pcpvt-s_uperhead_8x4_512x512_160k_ade20k.py │ │ │ ├── twins_svt-b_fpn_fpnhead_8x4_512x512_80k_ade20k.py │ │ │ ├── twins_svt-b_uperhead_8x2_512x512_160k_ade20k.py │ │ │ ├── twins_svt-l_fpn_fpnhead_8x4_512x512_80k_ade20k.py │ │ │ ├── twins_svt-l_uperhead_8x2_512x512_160k_ade20k.py │ │ │ ├── twins_svt-s_fpn_fpnhead_8x4_512x512_80k_ade20k.py │ │ │ └── twins_svt-s_uperhead_8x2_512x512_160k_ade20k.py │ │ ├── unet │ │ │ ├── README.md │ │ │ ├── deeplabv3_unet_s5-d16_128x128_40k_chase_db1.py │ │ │ ├── deeplabv3_unet_s5-d16_128x128_40k_stare.py │ │ │ ├── deeplabv3_unet_s5-d16_256x256_40k_hrf.py │ │ │ ├── deeplabv3_unet_s5-d16_64x64_40k_drive.py │ │ │ ├── deeplabv3_unet_s5-d16_ce-1.0-dice-3.0_128x128_40k_chase-db1.py │ │ │ ├── deeplabv3_unet_s5-d16_ce-1.0-dice-3.0_128x128_40k_stare.py │ │ │ ├── deeplabv3_unet_s5-d16_ce-1.0-dice-3.0_256x256_40k_hrf.py │ │ │ ├── deeplabv3_unet_s5-d16_ce-1.0-dice-3.0_64x64_40k_drive.py │ │ │ ├── fcn_unet_s5-d16_128x128_40k_chase_db1.py │ │ │ ├── fcn_unet_s5-d16_128x128_40k_stare.py │ │ │ ├── fcn_unet_s5-d16_256x256_40k_hrf.py │ │ │ ├── fcn_unet_s5-d16_4x4_512x1024_160k_cityscapes.py │ │ │ ├── fcn_unet_s5-d16_64x64_40k_drive.py │ │ │ ├── fcn_unet_s5-d16_ce-1.0-dice-3.0_128x128_40k_chase-db1.py │ │ │ ├── fcn_unet_s5-d16_ce-1.0-dice-3.0_128x128_40k_stare.py │ │ │ ├── fcn_unet_s5-d16_ce-1.0-dice-3.0_256x256_40k_hrf.py │ │ │ ├── fcn_unet_s5-d16_ce-1.0-dice-3.0_64x64_40k_drive.py │ │ │ ├── pspnet_unet_s5-d16_128x128_40k_chase_db1.py │ │ │ ├── pspnet_unet_s5-d16_128x128_40k_stare.py │ │ │ ├── pspnet_unet_s5-d16_256x256_40k_hrf.py │ │ │ ├── pspnet_unet_s5-d16_64x64_40k_drive.py │ │ │ ├── pspnet_unet_s5-d16_ce-1.0-dice-3.0_128x128_40k_chase-db1.py │ │ │ ├── pspnet_unet_s5-d16_ce-1.0-dice-3.0_128x128_40k_stare.py │ │ │ ├── pspnet_unet_s5-d16_ce-1.0-dice-3.0_256x256_40k_hrf.py │ │ │ ├── pspnet_unet_s5-d16_ce-1.0-dice-3.0_64x64_40k_drive.py │ │ │ └── unet.yml │ │ ├── upernet │ │ │ ├── README.md │ │ │ ├── upernet.yml │ │ │ ├── upernet_r101_512x1024_40k_cityscapes.py │ │ │ ├── upernet_r101_512x1024_80k_cityscapes.py │ │ │ ├── upernet_r101_512x512_160k_ade20k.py │ │ │ ├── upernet_r101_512x512_20k_voc12aug.py │ │ │ ├── upernet_r101_512x512_40k_voc12aug.py │ │ │ ├── upernet_r101_512x512_80k_ade20k.py │ │ │ ├── upernet_r101_769x769_40k_cityscapes.py │ │ │ ├── upernet_r101_769x769_80k_cityscapes.py │ │ │ ├── upernet_r50_512x1024_40k_cityscapes.py │ │ │ ├── upernet_r50_512x1024_80k_cityscapes.py │ │ │ ├── upernet_r50_512x512_160k_ade20k.py │ │ │ ├── upernet_r50_512x512_20k_voc12aug.py │ │ │ ├── upernet_r50_512x512_40k_voc12aug.py │ │ │ ├── upernet_r50_512x512_80k_ade20k.py │ │ │ ├── upernet_r50_769x769_40k_cityscapes.py │ │ │ └── upernet_r50_769x769_80k_cityscapes.py │ │ └── vit │ │ │ ├── README.md │ │ │ ├── upernet_deit-b16_512x512_160k_ade20k.py │ │ │ ├── upernet_deit-b16_512x512_80k_ade20k.py │ │ │ ├── upernet_deit-b16_ln_mln_512x512_160k_ade20k.py │ │ │ ├── upernet_deit-b16_mln_512x512_160k_ade20k.py │ │ │ ├── upernet_deit-s16_512x512_160k_ade20k.py │ │ │ ├── upernet_deit-s16_512x512_80k_ade20k.py │ │ │ ├── upernet_deit-s16_ln_mln_512x512_160k_ade20k.py │ │ │ ├── upernet_deit-s16_mln_512x512_160k_ade20k.py │ │ │ ├── upernet_vit-b16_ln_mln_512x512_160k_ade20k.py │ │ │ ├── upernet_vit-b16_mln_512x512_160k_ade20k.py │ │ │ ├── upernet_vit-b16_mln_512x512_80k_ade20k.py │ │ │ └── vit.yml │ ├── demo │ │ ├── image_demo.py │ │ └── video_demo.py │ ├── docker │ │ ├── Dockerfile │ │ └── serve │ │ │ ├── Dockerfile │ │ │ ├── config.properties │ │ │ └── entrypoint.sh │ ├── docs │ │ ├── en │ │ │ ├── Makefile │ │ │ ├── _static │ │ │ │ └── css │ │ │ │ │ └── readthedocs.css │ │ │ ├── api.rst │ │ │ ├── changelog.md │ │ │ ├── conf.py │ │ │ ├── dataset_prepare.md │ │ │ ├── get_started.md │ │ │ ├── index.rst │ │ │ ├── inference.md │ │ │ ├── make.bat │ │ │ ├── model_zoo.md │ │ │ ├── stat.py │ │ │ ├── switch_language.md │ │ │ ├── train.md │ │ │ ├── tutorials │ │ │ │ ├── config.md │ │ │ │ ├── customize_datasets.md │ │ │ │ ├── customize_models.md │ │ │ │ ├── customize_runtime.md │ │ │ │ ├── data_pipeline.md │ │ │ │ ├── index.rst │ │ │ │ └── training_tricks.md │ │ │ └── useful_tools.md │ │ └── zh_cn │ │ │ ├── Makefile │ │ │ ├── _static │ │ │ └── css │ │ │ │ └── readthedocs.css │ │ │ ├── api.rst │ │ │ ├── conf.py │ │ │ ├── dataset_prepare.md │ │ │ ├── get_started.md │ │ │ ├── index.rst │ │ │ ├── inference.md │ │ │ ├── make.bat │ │ │ ├── model_zoo.md │ │ │ ├── stat.py │ │ │ ├── switch_language.md │ │ │ ├── train.md │ │ │ ├── tutorials │ │ │ ├── config.md │ │ │ ├── customize_datasets.md │ │ │ ├── customize_models.md │ │ │ ├── customize_runtime.md │ │ │ ├── data_pipeline.md │ │ │ ├── index.rst │ │ │ └── training_tricks.md │ │ │ └── useful_tools.md │ ├── mmseg │ │ ├── .mim │ │ │ ├── configs │ │ │ ├── model-index.yml │ │ │ └── tools │ │ ├── __init__.py │ │ ├── apis │ │ │ ├── __init__.py │ │ │ ├── inference.py │ │ │ ├── test.py │ │ │ └── train.py │ │ ├── core │ │ │ ├── __init__.py │ │ │ ├── evaluation │ │ │ │ ├── __init__.py │ │ │ │ ├── class_names.py │ │ │ │ ├── eval_hooks.py │ │ │ │ └── metrics.py │ │ │ ├── seg │ │ │ │ ├── __init__.py │ │ │ │ ├── builder.py │ │ │ │ └── sampler │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── base_pixel_sampler.py │ │ │ │ │ └── ohem_pixel_sampler.py │ │ │ └── utils │ │ │ │ ├── __init__.py │ │ │ │ ├── layer_decay_optimizer_constructor.py │ │ │ │ └── misc.py │ │ ├── datasets │ │ │ ├── __init__.py │ │ │ ├── ade.py │ │ │ ├── builder.py │ │ │ ├── chase_db1.py │ │ │ ├── cityscapes.py │ │ │ ├── coco_stuff.py │ │ │ ├── custom.py │ │ │ ├── dark_zurich.py │ │ │ ├── dataset_wrappers.py │ │ │ ├── drive.py │ │ │ ├── hrf.py │ │ │ ├── isaid.py │ │ │ ├── isprs.py │ │ │ ├── loveda.py │ │ │ ├── night_driving.py │ │ │ ├── pascal_context.py │ │ │ ├── pipelines │ │ │ │ ├── __init__.py │ │ │ │ ├── compose.py │ │ │ │ ├── formating.py │ │ │ │ ├── formatting.py │ │ │ │ ├── loading.py │ │ │ │ ├── test_time_aug.py │ │ │ │ └── transforms.py │ │ │ ├── potsdam.py │ │ │ ├── stare.py │ │ │ └── voc.py │ │ ├── models │ │ │ ├── __init__.py │ │ │ ├── backbones │ │ │ │ ├── __init__.py │ │ │ │ ├── bisenetv1.py │ │ │ │ ├── bisenetv2.py │ │ │ │ ├── cgnet.py │ │ │ │ ├── erfnet.py │ │ │ │ ├── fast_scnn.py │ │ │ │ ├── hrnet.py │ │ │ │ ├── icnet.py │ │ │ │ ├── mit.py │ │ │ │ ├── mobilenet_v2.py │ │ │ │ ├── mobilenet_v3.py │ │ │ │ ├── resnest.py │ │ │ │ ├── resnet.py │ │ │ │ ├── resnext.py │ │ │ │ ├── stdc.py │ │ │ │ ├── swin.py │ │ │ │ ├── timm_backbone.py │ │ │ │ ├── twins.py │ │ │ │ ├── unet.py │ │ │ │ └── vit.py │ │ │ ├── builder.py │ │ │ ├── decode_heads │ │ │ │ ├── __init__.py │ │ │ │ ├── ann_head.py │ │ │ │ ├── apc_head.py │ │ │ │ ├── aspp_head.py │ │ │ │ ├── cascade_decode_head.py │ │ │ │ ├── cc_head.py │ │ │ │ ├── da_head.py │ │ │ │ ├── decode_head.py │ │ │ │ ├── dm_head.py │ │ │ │ ├── dnl_head.py │ │ │ │ ├── dpt_head.py │ │ │ │ ├── ema_head.py │ │ │ │ ├── enc_head.py │ │ │ │ ├── fcn_head.py │ │ │ │ ├── fpn_head.py │ │ │ │ ├── gc_head.py │ │ │ │ ├── isa_head.py │ │ │ │ ├── lraspp_head.py │ │ │ │ ├── nl_head.py │ │ │ │ ├── ocr_head.py │ │ │ │ ├── point_head.py │ │ │ │ ├── psa_head.py │ │ │ │ ├── psp_head.py │ │ │ │ ├── segformer_head.py │ │ │ │ ├── segmenter_mask_head.py │ │ │ │ ├── sep_aspp_head.py │ │ │ │ ├── sep_fcn_head.py │ │ │ │ ├── setr_mla_head.py │ │ │ │ ├── setr_up_head.py │ │ │ │ ├── stdc_head.py │ │ │ │ └── uper_head.py │ │ │ ├── losses │ │ │ │ ├── __init__.py │ │ │ │ ├── accuracy.py │ │ │ │ ├── cross_entropy_loss.py │ │ │ │ ├── dice_loss.py │ │ │ │ ├── focal_loss.py │ │ │ │ ├── lovasz_loss.py │ │ │ │ └── utils.py │ │ │ ├── necks │ │ │ │ ├── __init__.py │ │ │ │ ├── fpn.py │ │ │ │ ├── ic_neck.py │ │ │ │ ├── jpu.py │ │ │ │ ├── mla_neck.py │ │ │ │ └── multilevel_neck.py │ │ │ ├── segmentors │ │ │ │ ├── __init__.py │ │ │ │ ├── base.py │ │ │ │ ├── cascade_encoder_decoder.py │ │ │ │ └── encoder_decoder.py │ │ │ └── utils │ │ │ │ ├── __init__.py │ │ │ │ ├── embed.py │ │ │ │ ├── inverted_residual.py │ │ │ │ ├── make_divisible.py │ │ │ │ ├── res_layer.py │ │ │ │ ├── se_layer.py │ │ │ │ ├── self_attention_block.py │ │ │ │ ├── shape_convert.py │ │ │ │ └── up_conv_block.py │ │ ├── ops │ │ │ ├── __init__.py │ │ │ ├── encoding.py │ │ │ └── wrappers.py │ │ ├── utils │ │ │ ├── __init__.py │ │ │ ├── collect_env.py │ │ │ ├── logger.py │ │ │ ├── misc.py │ │ │ └── set_env.py │ │ └── version.py │ ├── model-index.yml │ ├── pytest.ini │ ├── requirements.txt │ ├── requirements │ │ ├── docs.txt │ │ ├── mminstall.txt │ │ ├── optional.txt │ │ ├── readthedocs.txt │ │ ├── runtime.txt │ │ └── tests.txt │ ├── resources │ │ └── seg_demo.gif │ ├── setup.cfg │ ├── setup.py │ ├── tests │ │ ├── __init__.py │ │ ├── test_apis │ │ │ └── test_single_gpu.py │ │ ├── test_config.py │ │ ├── test_core │ │ │ └── test_learning_rate_decay_optimizer_constructor.py │ │ ├── test_data │ │ │ ├── test_dataset.py │ │ │ ├── test_dataset_builder.py │ │ │ ├── test_loading.py │ │ │ ├── test_transform.py │ │ │ └── test_tta.py │ │ ├── test_digit_version.py │ │ ├── test_eval_hook.py │ │ ├── test_inference.py │ │ ├── test_metrics.py │ │ ├── test_models │ │ │ ├── __init__.py │ │ │ ├── test_backbones │ │ │ │ ├── __init__.py │ │ │ │ ├── test_bisenetv1.py │ │ │ │ ├── test_bisenetv2.py │ │ │ │ ├── test_blocks.py │ │ │ │ ├── test_cgnet.py │ │ │ │ ├── test_erfnet.py │ │ │ │ ├── test_fast_scnn.py │ │ │ │ ├── test_hrnet.py │ │ │ │ ├── test_icnet.py │ │ │ │ ├── test_mit.py │ │ │ │ ├── test_mobilenet_v3.py │ │ │ │ ├── test_resnest.py │ │ │ │ ├── test_resnet.py │ │ │ │ ├── test_resnext.py │ │ │ │ ├── test_stdc.py │ │ │ │ ├── test_swin.py │ │ │ │ ├── test_timm_backbone.py │ │ │ │ ├── test_twins.py │ │ │ │ ├── test_unet.py │ │ │ │ ├── test_vit.py │ │ │ │ └── utils.py │ │ │ ├── test_forward.py │ │ │ ├── test_heads │ │ │ │ ├── __init__.py │ │ │ │ ├── test_ann_head.py │ │ │ │ ├── test_apc_head.py │ │ │ │ ├── test_aspp_head.py │ │ │ │ ├── test_cc_head.py │ │ │ │ ├── test_da_head.py │ │ │ │ ├── test_decode_head.py │ │ │ │ ├── test_dm_head.py │ │ │ │ ├── test_dnl_head.py │ │ │ │ ├── test_dpt_head.py │ │ │ │ ├── test_ema_head.py │ │ │ │ ├── test_enc_head.py │ │ │ │ ├── test_fcn_head.py │ │ │ │ ├── test_gc_head.py │ │ │ │ ├── test_isa_head.py │ │ │ │ ├── test_lraspp_head.py │ │ │ │ ├── test_nl_head.py │ │ │ │ ├── test_ocr_head.py │ │ │ │ ├── test_point_head.py │ │ │ │ ├── test_psa_head.py │ │ │ │ ├── test_psp_head.py │ │ │ │ ├── test_segformer_head.py │ │ │ │ ├── test_segmenter_mask_head.py │ │ │ │ ├── test_setr_mla_head.py │ │ │ │ ├── test_setr_up_head.py │ │ │ │ ├── test_stdc_head.py │ │ │ │ ├── test_uper_head.py │ │ │ │ └── utils.py │ │ │ ├── test_losses │ │ │ │ ├── __init__.py │ │ │ │ ├── test_ce_loss.py │ │ │ │ ├── test_dice_loss.py │ │ │ │ ├── test_focal_loss.py │ │ │ │ ├── test_lovasz_loss.py │ │ │ │ └── test_utils.py │ │ │ ├── test_necks │ │ │ │ ├── __init__.py │ │ │ │ ├── test_fpn.py │ │ │ │ ├── test_ic_neck.py │ │ │ │ ├── test_jpu.py │ │ │ │ ├── test_mla_neck.py │ │ │ │ └── test_multilevel_neck.py │ │ │ ├── test_segmentors │ │ │ │ ├── __init__.py │ │ │ │ ├── test_cascade_encoder_decoder.py │ │ │ │ ├── test_encoder_decoder.py │ │ │ │ └── utils.py │ │ │ └── test_utils │ │ │ │ ├── __init__.py │ │ │ │ └── test_embed.py │ │ ├── test_sampler.py │ │ └── test_utils │ │ │ ├── test_misc.py │ │ │ └── test_set_env.py │ └── tools │ │ ├── analyze_logs.py │ │ ├── benchmark.py │ │ ├── browse_dataset.py │ │ ├── confusion_matrix.py │ │ ├── convert_datasets │ │ ├── chase_db1.py │ │ ├── cityscapes.py │ │ ├── coco_stuff10k.py │ │ ├── coco_stuff164k.py │ │ ├── drive.py │ │ ├── hrf.py │ │ ├── isaid.py │ │ ├── loveda.py │ │ ├── pascal_context.py │ │ ├── potsdam.py │ │ ├── stare.py │ │ ├── vaihingen.py │ │ └── voc_aug.py │ │ ├── deploy_test.py │ │ ├── dist_test.sh │ │ ├── dist_train.sh │ │ ├── get_flops.py │ │ ├── model_converters │ │ ├── mit2mmseg.py │ │ ├── stdc2mmseg.py │ │ ├── swin2mmseg.py │ │ ├── twins2mmseg.py │ │ ├── vit2mmseg.py │ │ └── vitjax2mmseg.py │ │ ├── onnx2tensorrt.py │ │ ├── print_config.py │ │ ├── publish_model.py │ │ ├── pytorch2onnx.py │ │ ├── pytorch2torchscript.py │ │ ├── slurm_test.sh │ │ ├── slurm_train.sh │ │ ├── test.py │ │ ├── torchserve │ │ ├── mmseg2torchserve.py │ │ ├── mmseg_handler.py │ │ └── test_torchserve.py │ │ └── train.py └── mmtracking │ ├── .circleci │ └── config.yml │ ├── .dev_scripts │ └── benchmark │ │ ├── batch_train_list.txt │ │ ├── gather_train_benchmark_metric.py │ │ └── train_benchmark.sh │ ├── .pre-commit-config.yaml │ ├── .readthedocs.yml │ ├── CITATION.cff │ ├── LICENSE │ ├── MANIFEST.in │ ├── README.md │ ├── README_zh-CN.md │ ├── configs │ ├── README.md │ ├── _base_ │ │ ├── datasets │ │ │ ├── coco_detection.py │ │ │ ├── coco_instance.py │ │ │ ├── coco_instance_semantic.py │ │ │ ├── dancetrack.py │ │ │ ├── imagenet_vid_dff_style.py │ │ │ ├── imagenet_vid_fgfa_style.py │ │ │ ├── lvis_v0.5_instance.py │ │ │ ├── lvis_v1_instance.py │ │ │ ├── mot_challenge.py │ │ │ ├── mot_challenge_det.py │ │ │ ├── mot_challenge_reid.py │ │ │ ├── tao.py │ │ │ └── youtube_vis.py │ │ ├── default_runtime.py │ │ └── models │ │ │ ├── cascade_mask_rcnn_r50_fpn.py │ │ │ ├── cascade_rcnn_r50_fpn.py │ │ │ ├── faster_rcnn_r50_caffe_c4.py │ │ │ ├── faster_rcnn_r50_dc5.py │ │ │ ├── faster_rcnn_r50_fpn.py │ │ │ ├── mask_rcnn_r50_fpn.py │ │ │ ├── retinanet_r50_fpn.py │ │ │ └── yolox_x_8x8.py │ ├── det │ │ ├── faster-rcnn_r101_fpn_4e_mot17-half.py │ │ ├── faster-rcnn_r50_fpn_4e_mot15-half.py │ │ ├── faster-rcnn_r50_fpn_4e_mot16-half.py │ │ ├── faster-rcnn_r50_fpn_4e_mot17-half.py │ │ ├── faster-rcnn_r50_fpn_4e_mot17.py │ │ ├── faster-rcnn_r50_fpn_8e_mot20-half.py │ │ └── faster-rcnn_r50_fpn_8e_mot20.py │ ├── mot │ │ ├── README.md │ │ ├── bytetrack │ │ │ ├── README.md │ │ │ ├── bytetrack_yolox_x_crowdhuman_mot17-private-half.py │ │ │ ├── bytetrack_yolox_x_crowdhuman_mot17-private.py │ │ │ ├── bytetrack_yolox_x_crowdhuman_mot20-private.py │ │ │ └── metafile.yml │ │ ├── deepsort │ │ │ ├── README.md │ │ │ ├── deepsort_faster-rcnn_fpn_4e_mot17-private-half.py │ │ │ ├── deepsort_faster-rcnn_fpn_4e_mot17-public-half.py │ │ │ ├── metafile.yml │ │ │ ├── sort_faster-rcnn_fpn_4e_mot17-private-half.py │ │ │ ├── sort_faster-rcnn_fpn_4e_mot17-private.py │ │ │ ├── sort_faster-rcnn_fpn_4e_mot17-public-half.py │ │ │ └── sort_faster-rcnn_fpn_4e_mot17-public.py │ │ ├── qdtrack │ │ │ ├── README.md │ │ │ ├── metafile.yml │ │ │ ├── qdtrack_faster-rcnn_r101_fpn_12e_tao.py │ │ │ ├── qdtrack_faster-rcnn_r101_fpn_24e_lvis.py │ │ │ ├── qdtrack_faster-rcnn_r50_fpn_4e_base.py │ │ │ ├── qdtrack_faster-rcnn_r50_fpn_4e_crowdhuman_mot17-private-half.py │ │ │ ├── qdtrack_faster-rcnn_r50_fpn_4e_dancetrack.py │ │ │ └── qdtrack_faster-rcnn_r50_fpn_4e_mot17-private-half.py │ │ └── tracktor │ │ │ ├── README.md │ │ │ ├── metafile.yml │ │ │ ├── tracktor_faster-rcnn_r50_fpn_4e_mot15-private-half.py │ │ │ ├── tracktor_faster-rcnn_r50_fpn_4e_mot15-public-half.py │ │ │ ├── tracktor_faster-rcnn_r50_fpn_4e_mot16-private-half.py │ │ │ ├── tracktor_faster-rcnn_r50_fpn_4e_mot16-public-half.py │ │ │ ├── tracktor_faster-rcnn_r50_fpn_4e_mot17-private-half.py │ │ │ ├── tracktor_faster-rcnn_r50_fpn_4e_mot17-private.py │ │ │ ├── tracktor_faster-rcnn_r50_fpn_4e_mot17-public-half.py │ │ │ ├── tracktor_faster-rcnn_r50_fpn_4e_mot17-public-half_search.py │ │ │ ├── tracktor_faster-rcnn_r50_fpn_4e_mot17-public.py │ │ │ ├── tracktor_faster-rcnn_r50_fpn_8e_mot20-private-half.py │ │ │ ├── tracktor_faster-rcnn_r50_fpn_8e_mot20-public-half.py │ │ │ ├── tracktor_faster-rcnn_r50_fpn_8e_mot20-public.py │ │ │ └── tracktor_faster-rcnn_r50_fpn_fp16_4e_mot17-private-half.py │ ├── reid │ │ ├── resnet50_b32x8_MOT15.py │ │ ├── resnet50_b32x8_MOT16.py │ │ ├── resnet50_b32x8_MOT17.py │ │ └── resnet50_b32x8_MOT20.py │ ├── sot │ │ ├── README.md │ │ ├── siamese_rpn │ │ │ ├── README.md │ │ │ ├── metafile.yml │ │ │ ├── siamese_rpn_r50_20e_lasot.py │ │ │ ├── siamese_rpn_r50_20e_otb100.py │ │ │ ├── siamese_rpn_r50_20e_trackingnet.py │ │ │ ├── siamese_rpn_r50_20e_uav123.py │ │ │ ├── siamese_rpn_r50_20e_vot2018.py │ │ │ └── siamese_rpn_r50_fp16_20e_lasot.py │ │ └── stark │ │ │ ├── README.md │ │ │ ├── metafile.yml │ │ │ ├── stark_st1_r50_500e_got10k.py │ │ │ ├── stark_st1_r50_500e_lasot.py │ │ │ ├── stark_st1_r50_500e_trackingnet.py │ │ │ ├── stark_st2_r50_50e_got10k.py │ │ │ ├── stark_st2_r50_50e_lasot.py │ │ │ └── stark_st2_r50_50e_trackingnet.py │ └── vid │ │ ├── README.md │ │ ├── dff │ │ ├── README.md │ │ ├── dff_faster_rcnn_r101_dc5_1x_imagenetvid.py │ │ ├── dff_faster_rcnn_r50_dc5_1x_imagenetvid.py │ │ ├── dff_faster_rcnn_x101_dc5_1x_imagenetvid.py │ │ └── metafile.yml │ │ ├── fgfa │ │ ├── README.md │ │ ├── fgfa_faster_rcnn_r101_dc5_1x_imagenetvid.py │ │ ├── fgfa_faster_rcnn_r50_dc5_1x_imagenetvid.py │ │ ├── fgfa_faster_rcnn_x101_dc5_1x_imagenetvid.py │ │ └── metafile.yml │ │ ├── selsa │ │ ├── README.md │ │ ├── metafile.yml │ │ ├── selsa_faster_rcnn_r101_dc5_1x_imagenetvid.py │ │ ├── selsa_faster_rcnn_r50_dc5_1x_imagenetvid.py │ │ ├── selsa_faster_rcnn_r50_dc5_fp16_1x_imagenetvid.py │ │ └── selsa_faster_rcnn_x101_dc5_1x_imagenetvid.py │ │ └── temporal_roi_align │ │ ├── README.md │ │ ├── metafile.yml │ │ ├── selsa_troialign_faster_rcnn_r101_dc5_7e_imagenetvid.py │ │ ├── selsa_troialign_faster_rcnn_r50_dc5_7e_imagenetvid.py │ │ └── selsa_troialign_faster_rcnn_x101_dc5_7e_imagenetvid.py │ ├── demo │ ├── README.md │ ├── demo.mp4 │ ├── demo_mot_vis.py │ ├── demo_sot.py │ └── demo_vid.py │ ├── docker │ └── Dockerfile │ ├── docs │ ├── en │ │ ├── Makefile │ │ ├── _static │ │ │ └── css │ │ │ │ └── readthedocs.css │ │ ├── api.rst │ │ ├── changelog.md │ │ ├── conf.py │ │ ├── dataset.md │ │ ├── index.rst │ │ ├── install.md │ │ ├── make.bat │ │ ├── model_zoo.md │ │ ├── quick_run.md │ │ ├── stat.py │ │ ├── switch_language.md │ │ ├── tutorials │ │ │ ├── config.md │ │ │ ├── config_mot.md │ │ │ ├── config_sot.md │ │ │ ├── config_vid.md │ │ │ ├── customize_data_pipeline.md │ │ │ ├── customize_dataset.md │ │ │ ├── customize_mot_model.md │ │ │ ├── customize_runtime.md │ │ │ ├── customize_sot_model.md │ │ │ ├── customize_vid_model.md │ │ │ └── index.rst │ │ └── useful_tools_scripts.md │ └── zh_cn │ │ ├── Makefile │ │ ├── _static │ │ └── css │ │ │ └── readthedocs.css │ │ ├── api.rst │ │ ├── changelog.md │ │ ├── conf.py │ │ ├── dataset.md │ │ ├── index.rst │ │ ├── install.md │ │ ├── make.bat │ │ ├── model_zoo.md │ │ ├── quick_run.md │ │ ├── stat.py │ │ ├── switch_language.md │ │ ├── tutorials │ │ ├── config.md │ │ ├── config_mot.md │ │ ├── config_sot.md │ │ ├── config_vid.md │ │ ├── customize_data_pipeline.md │ │ ├── customize_dataset.md │ │ ├── customize_mot_model.md │ │ ├── customize_runtime.md │ │ ├── customize_sot_model.md │ │ ├── customize_vid_model.md │ │ └── index.rst │ │ └── useful_tools_scripts.md │ ├── mmtrack │ ├── .mim │ │ ├── configs │ │ ├── model-index.yml │ │ └── tools │ ├── __init__.py │ ├── apis │ │ ├── __init__.py │ │ ├── inference.py │ │ ├── test.py │ │ └── train.py │ ├── core │ │ ├── __init__.py │ │ ├── anchor │ │ │ ├── __init__.py │ │ │ └── sot_anchor_generator.py │ │ ├── bbox │ │ │ ├── __init__.py │ │ │ ├── iou_calculators │ │ │ │ ├── __init__.py │ │ │ │ └── region_iou_calculator.py │ │ │ └── transforms.py │ │ ├── evaluation │ │ │ ├── __init__.py │ │ │ ├── eval_hooks.py │ │ │ ├── eval_mot.py │ │ │ ├── eval_sot_ope.py │ │ │ ├── eval_sot_vot.py │ │ │ ├── eval_vis.py │ │ │ ├── ytvis.py │ │ │ └── ytviseval.py │ │ ├── hook │ │ │ ├── __init__.py │ │ │ └── yolox_mode_switch_hook.py │ │ ├── motion │ │ │ ├── __init__.py │ │ │ └── flow.py │ │ ├── optimizer │ │ │ ├── __init__.py │ │ │ ├── sot_lr_updater.py │ │ │ └── sot_optimizer_hook.py │ │ ├── track │ │ │ ├── __init__.py │ │ │ ├── correlation.py │ │ │ ├── interpolation.py │ │ │ ├── similarity.py │ │ │ └── transforms.py │ │ └── utils │ │ │ ├── __init__.py │ │ │ ├── image.py │ │ │ ├── misc.py │ │ │ └── visualization.py │ ├── datasets │ │ ├── __init__.py │ │ ├── base_sot_dataset.py │ │ ├── builder.py │ │ ├── coco_video_dataset.py │ │ ├── dancetrack_dataset.py │ │ ├── dataset_wrappers.py │ │ ├── got10k_dataset.py │ │ ├── imagenet_vid_dataset.py │ │ ├── lasot_dataset.py │ │ ├── mot_challenge_dataset.py │ │ ├── otb_dataset.py │ │ ├── parsers │ │ │ ├── __init__.py │ │ │ └── coco_video_parser.py │ │ ├── pipelines │ │ │ ├── __init__.py │ │ │ ├── formatting.py │ │ │ ├── loading.py │ │ │ ├── processing.py │ │ │ └── transforms.py │ │ ├── reid_dataset.py │ │ ├── samplers │ │ │ ├── __init__.py │ │ │ ├── quota_sampler.py │ │ │ └── video_sampler.py │ │ ├── sot_coco_dataset.py │ │ ├── sot_imagenet_vid_dataset.py │ │ ├── sot_test_dataset.py │ │ ├── sot_train_dataset.py │ │ ├── tao_dataset.py │ │ ├── trackingnet_dataset.py │ │ ├── uav123_dataset.py │ │ ├── vot_dataset.py │ │ └── youtube_vis_dataset.py │ ├── models │ │ ├── __init__.py │ │ ├── aggregators │ │ │ ├── __init__.py │ │ │ ├── embed_aggregator.py │ │ │ └── selsa_aggregator.py │ │ ├── backbones │ │ │ ├── __init__.py │ │ │ └── sot_resnet.py │ │ ├── builder.py │ │ ├── losses │ │ │ ├── __init__.py │ │ │ ├── l2_loss.py │ │ │ ├── multipos_cross_entropy_loss.py │ │ │ └── triplet_loss.py │ │ ├── mot │ │ │ ├── __init__.py │ │ │ ├── base.py │ │ │ ├── byte_track.py │ │ │ ├── deep_sort.py │ │ │ ├── qdtrack.py │ │ │ └── tracktor.py │ │ ├── motion │ │ │ ├── __init__.py │ │ │ ├── camera_motion_compensation.py │ │ │ ├── flownet_simple.py │ │ │ ├── kalman_filter.py │ │ │ └── linear_motion.py │ │ ├── reid │ │ │ ├── __init__.py │ │ │ ├── base_reid.py │ │ │ ├── fc_module.py │ │ │ ├── gap.py │ │ │ └── linear_reid_head.py │ │ ├── roi_heads │ │ │ ├── __init__.py │ │ │ ├── bbox_heads │ │ │ │ ├── __init__.py │ │ │ │ └── selsa_bbox_head.py │ │ │ ├── roi_extractors │ │ │ │ ├── __init__.py │ │ │ │ ├── single_level_roi_extractor.py │ │ │ │ └── temporal_roi_align.py │ │ │ └── selsa_roi_head.py │ │ ├── sot │ │ │ ├── __init__.py │ │ │ ├── base.py │ │ │ ├── siamrpn.py │ │ │ └── stark.py │ │ ├── track_heads │ │ │ ├── __init__.py │ │ │ ├── quasi_dense_embed_head.py │ │ │ ├── quasi_dense_track_head.py │ │ │ ├── roi_embed_head.py │ │ │ ├── roi_track_head.py │ │ │ ├── siamese_rpn_head.py │ │ │ └── stark_head.py │ │ ├── trackers │ │ │ ├── __init__.py │ │ │ ├── base_tracker.py │ │ │ ├── byte_tracker.py │ │ │ ├── masktrack_rcnn_tracker.py │ │ │ ├── quasi_dense_tao_tracker.py │ │ │ ├── quasi_dense_tracker.py │ │ │ ├── sort_tracker.py │ │ │ └── tracktor_tracker.py │ │ └── vid │ │ │ ├── __init__.py │ │ │ ├── base.py │ │ │ ├── dff.py │ │ │ ├── fgfa.py │ │ │ └── selsa.py │ ├── utils │ │ ├── __init__.py │ │ ├── collect_env.py │ │ └── logger.py │ └── version.py │ ├── model-index.yml │ ├── requirements.txt │ ├── requirements │ ├── build.txt │ ├── docs.txt │ ├── mminstall.txt │ ├── readthedocs.txt │ ├── runtime.txt │ └── tests.txt │ ├── setup.cfg │ ├── setup.py │ ├── tests │ ├── test_core │ │ ├── test_bbox │ │ │ └── test_bbox_transforms.py │ │ ├── test_motion │ │ │ └── test_flow.py │ │ ├── test_track │ │ │ ├── test_interpolation.py │ │ │ ├── test_similarity.py │ │ │ └── test_transforms.py │ │ └── test_utils │ │ │ └── test_visualization.py │ ├── test_data │ │ ├── test_datasets │ │ │ ├── __init__.py │ │ │ ├── test_coco_video_dataset.py │ │ │ ├── test_common.py │ │ │ ├── test_dataset_wrapers.py │ │ │ ├── test_mot_challenge_dataset.py │ │ │ ├── test_reid_dataset.py │ │ │ ├── test_sot_dataset.py │ │ │ ├── test_sot_train_dataset.py │ │ │ ├── test_tao_dataset.py │ │ │ ├── test_vis_dataset.py │ │ │ └── utils.py │ │ └── test_pipelines │ │ │ ├── test_formatting.py │ │ │ ├── test_loading.py │ │ │ ├── test_processing.py │ │ │ └── test_transform.py │ ├── test_models │ │ ├── test_aggregators │ │ │ ├── test_embed_aggregator.py │ │ │ └── test_selsa_aggregator.py │ │ ├── test_backones │ │ │ └── test_sot_resnet.py │ │ ├── test_forward │ │ │ ├── __init__.py │ │ │ ├── test_mot_forward.py │ │ │ ├── test_sot_forward.py │ │ │ ├── test_vid_forward.py │ │ │ ├── test_vis_forward.py │ │ │ └── utils.py │ │ ├── test_losses │ │ │ ├── test_multi_pos_cross_entropy_loss.py │ │ │ └── test_triplet_loss.py │ │ ├── test_motion │ │ │ ├── test_camera_motion_compensation.py │ │ │ ├── test_flownet_simple.py │ │ │ └── test_linear_motion.py │ │ ├── test_reid │ │ │ └── test_base_reid.py │ │ ├── test_roi_heads │ │ │ ├── test_bbox_heads │ │ │ │ └── test_selsa_bbox_head.py │ │ │ └── test_roi_extractors │ │ │ │ ├── test_single_level_roi_extractor.py │ │ │ │ └── test_temporal_roi_align.py │ │ ├── test_track_heads │ │ │ ├── test_quasi_dense_embed_head.py │ │ │ ├── test_roi_embed_head.py │ │ │ ├── test_siamese_rpn_head.py │ │ │ └── test_stark_head.py │ │ └── test_trackers │ │ │ ├── test_base_tracker.py │ │ │ ├── test_bytetracker.py │ │ │ ├── test_masktrack_rcnn_tracker.py │ │ │ ├── test_quasi_dense_embed_tracker.py │ │ │ └── test_quasi_dense_tao_trakcer.py │ └── test_version.py │ └── tools │ ├── analysis │ ├── analyze_logs.py │ ├── benchmark.py │ ├── mot │ │ ├── dist_mot_search.sh │ │ ├── mot_dummy_results.py │ │ ├── mot_error_visualize.py │ │ ├── mot_param_search.py │ │ └── slurm_mot_search.sh │ ├── print_config.py │ ├── publish_model.py │ └── sot │ │ ├── dist_sot_siamrpn_search.sh │ │ ├── slurm_sot_siamrpn_search.sh │ │ └── sot_siamrpn_param_search.py │ ├── convert_datasets │ ├── dancetrack │ │ └── dancetrack2coco.py │ ├── got10k │ │ ├── gen_got10k_infos.py │ │ ├── got10k2coco.py │ │ └── unzip_got10k.sh │ ├── ilsvrc │ │ ├── imagenet2coco_det.py │ │ └── imagenet2coco_vid.py │ ├── lasot │ │ ├── gen_lasot_infos.py │ │ ├── lasot2coco.py │ │ └── testing_set.txt │ ├── mot │ │ ├── crowdhuman2coco.py │ │ ├── mot2coco.py │ │ └── mot2reid.py │ ├── otb100 │ │ ├── download_otb100.py │ │ ├── otb100_infos.txt │ │ ├── otb2coco.py │ │ └── unzip_otb100.sh │ ├── tao │ │ ├── merge_coco_with_lvis.py │ │ └── tao2coco.py │ ├── trackingnet │ │ ├── gen_trackingnet_infos.py │ │ ├── trackingnet2coco.py │ │ └── unzip_trackinget.sh │ ├── uav123 │ │ ├── uav123_info_deprecated.txt │ │ ├── uav123_infos.txt │ │ └── uav2coco.py │ ├── vot │ │ ├── download_vot.py │ │ ├── gen_vot_infos.py │ │ └── vot2coco.py │ └── youtubevis │ │ └── youtubevis2coco.py │ ├── dist_test.sh │ ├── dist_train.sh │ ├── slurm_test.sh │ ├── slurm_train.sh │ ├── test.py │ ├── torchserve │ ├── mmtrack2torchserve.py │ ├── mmtrack_handler.py │ └── test_torchserve.py │ └── train.py └── tools ├── analysis_tools ├── analyze_logs.py ├── benchmark.py ├── get_flops.py ├── vis_nuscenes_bevdet.py └── vis_nuscenes_bevdet_tracking.py ├── check_exps.py ├── create_bevdet4d_info.py ├── create_data.py ├── create_data.sh ├── custom_slurm_test16.sh ├── custom_slurm_train.sh ├── custom_slurm_train16.sh ├── custom_slurm_traina.sh ├── data_converter ├── __init__.py ├── create_gt_database.py ├── indoor_converter.py ├── kitti_converter.py ├── kitti_data_utils.py ├── lyft_converter.py ├── lyft_data_fixer.py ├── nuimage_converter.py ├── nuscenes_converter.py ├── prepare_nuscenes_for_bevdet4d.py ├── s3dis_data_utils.py ├── scannet_data_utils.py ├── sunrgbd_data_utils.py └── waymo_converter.py ├── debug.sh ├── dist_eval_loss.sh ├── dist_test.sh ├── dist_train.sh ├── manager.py ├── misc ├── browse_dataset.py ├── browse_dataset_dev.py ├── fuse_conv_bn.py ├── print_config.py └── visualize_results.py ├── model_converters ├── convert_votenet_checkpoints.py ├── publish_model.py └── regnet2mmdet.py ├── slurm_test.sh ├── slurm_train.sh ├── test.py ├── test.sh └── train.py /build.sh: -------------------------------------------------------------------------------- 1 | cd software/mmdet/; 2 | python setup.py develop; 3 | cd ../mmdet3d/; 4 | python setup.py develop; 5 | cd ../mmseg/; 6 | python setup.py develop; 7 | cd ../../det3d/models/backbones/DCNv2_t18/; 8 | bash make.sh 9 | 10 | -------------------------------------------------------------------------------- /configs/_base_/models/paconv_cuda_ssg.py: -------------------------------------------------------------------------------- 1 | _base_ = './paconv_ssg.py' 2 | 3 | model = dict( 4 | backbone=dict( 5 | sa_cfg=dict( 6 | type='PAConvCUDASAModule', 7 | scorenet_cfg=dict(mlp_channels=[8, 16, 16])))) 8 | -------------------------------------------------------------------------------- /det3d/apis/__init__.py: -------------------------------------------------------------------------------- 1 | from .test import single_gpu_get_loss 2 | 3 | __all__ = ['single_gpu_get_loss'] 4 | -------------------------------------------------------------------------------- /det3d/core/bbox/assigners/__init__.py: -------------------------------------------------------------------------------- 1 | from .hungarian_assigner_3d import HungarianAssigner3D 2 | 3 | __all__ = ['HungarianAssigner3D'] 4 | -------------------------------------------------------------------------------- /det3d/core/bbox/coders/__init__.py: -------------------------------------------------------------------------------- 1 | from .nms_free_coder import NMSFreeCoder 2 | from .nocs_coder import NOCSCoder 3 | __all__ = ['NMSFreeCoder', 'NOCSCoder'] 4 | -------------------------------------------------------------------------------- /det3d/core/bbox/match_costs/__init__.py: -------------------------------------------------------------------------------- 1 | from mmdet.core.bbox.match_costs import build_match_cost 2 | from .match_cost import BBox3DL1Cost 3 | 4 | __all__ = ['build_match_cost', 'BBox3DL1Cost'] -------------------------------------------------------------------------------- /det3d/core/visualizer/__init__.py: -------------------------------------------------------------------------------- 1 | # from .show_result import show_custom_multi_modality_result 2 | 3 | # __all__ = ['show_custom_multi_modality_result'] -------------------------------------------------------------------------------- /det3d/hooks/__init__.py: -------------------------------------------------------------------------------- 1 | from .mlflow import CustomMlflowLoggerHook 2 | from .ema import ModelEMA 3 | from .sequentialcontrol import SequentialControlHook 4 | 5 | 6 | __all__ = ['CustomMlflowLoggerHook', 7 | 'ModelEMA', 'SequentialControlHook'] -------------------------------------------------------------------------------- /det3d/models/backbones/DCNv2/.gitignore: -------------------------------------------------------------------------------- 1 | .vscode 2 | .idea 3 | *.so 4 | *.o 5 | *pyc 6 | _ext 7 | build 8 | DCNv2.egg-info 9 | dist -------------------------------------------------------------------------------- /det3d/models/backbones/DCNv2/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenRobotLab/DORT/5c1f39259247813a5c340753dd1a274f253a39b7/det3d/models/backbones/DCNv2/__init__.py -------------------------------------------------------------------------------- /det3d/models/backbones/DCNv2/make.sh: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env bash 2 | rm *.so 3 | rm -r build/ 4 | python3 setup.py build develop 5 | -------------------------------------------------------------------------------- /det3d/models/backbones/DCNv2_t18/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenRobotLab/DORT/5c1f39259247813a5c340753dd1a274f253a39b7/det3d/models/backbones/DCNv2_t18/__init__.py -------------------------------------------------------------------------------- /det3d/models/backbones/DCNv2_t18/make.sh: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env bash 2 | rm *.so 3 | rm -r build/ 4 | python setup.py build develop 5 | -------------------------------------------------------------------------------- /det3d/models/losses/__init__.py: -------------------------------------------------------------------------------- 1 | from .uncertainty_l1_loss import UncertaintyL1Loss 2 | 3 | __all__ = ['UncertaintyL1Loss'] -------------------------------------------------------------------------------- /det3d/models/motion/__init__.py: -------------------------------------------------------------------------------- 1 | from .ab3dmot_kalman_filter import AB3DMOTKalmanFilter 2 | 3 | __all__ = ['AB3DMOTKalmanFilter'] -------------------------------------------------------------------------------- /det3d/models/track_frameworks/__init__.py: -------------------------------------------------------------------------------- 1 | from .qdtrack import CustomQDTrack 2 | from .sort import CustomSORT 3 | 4 | __all__ = ['CustomQDTrack', 'CustomSORT'] 5 | -------------------------------------------------------------------------------- /det3d/models/track_heads/__init__.py: -------------------------------------------------------------------------------- 1 | from .quasi_dense_track_head import CustomQuasiDenseTrackHead 2 | 3 | __all__ = ['CustomQuasiDenseTrackHead'] -------------------------------------------------------------------------------- /det3d/models/trackers/__init__.py: -------------------------------------------------------------------------------- 1 | from .quasi_dense_tracker import CustomQuasiDenseTracker 2 | from .sort_tracker import CustomSORTTracker 3 | 4 | __all__ = ['CustomQuasiDenseTracker', 'CustomSORTTracker'] 5 | -------------------------------------------------------------------------------- /det3d/ops/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenRobotLab/DORT/5c1f39259247813a5c340753dd1a274f253a39b7/det3d/ops/__init__.py -------------------------------------------------------------------------------- /det3d/ops/bev_pool_v2/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) Phigent Robotics. All rights reserved. 2 | -------------------------------------------------------------------------------- /det3d/version.py: -------------------------------------------------------------------------------- 1 | __version__ = "0.1.0+9647dac" 2 | -------------------------------------------------------------------------------- /figs/overview.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenRobotLab/DORT/5c1f39259247813a5c340753dd1a274f253a39b7/figs/overview.png -------------------------------------------------------------------------------- /software/mmdet/.dev_scripts/linter.sh: -------------------------------------------------------------------------------- 1 | yapf -r -i mmdet/ configs/ tests/ tools/ 2 | isort -rc mmdet/ configs/ tests/ tools/ 3 | flake8 . 4 | -------------------------------------------------------------------------------- /software/mmdet/.owners.yml: -------------------------------------------------------------------------------- 1 | assign: 2 | strategy: 3 | # random 4 | daily-shift-based 5 | scedule: 6 | '*/1 * * * *' 7 | assignees: 8 | - Czm369 9 | - hhaAndroid 10 | - jbwang1997 11 | - RangiLyu 12 | - BIGWangYuDong 13 | - chhluo 14 | - ZwwWayne 15 | -------------------------------------------------------------------------------- /software/mmdet/.readthedocs.yml: -------------------------------------------------------------------------------- 1 | version: 2 2 | 3 | formats: all 4 | 5 | python: 6 | version: 3.7 7 | install: 8 | - requirements: requirements/docs.txt 9 | - requirements: requirements/readthedocs.txt 10 | -------------------------------------------------------------------------------- /software/mmdet/CITATION.cff: -------------------------------------------------------------------------------- 1 | cff-version: 1.2.0 2 | message: "If you use this software, please cite it as below." 3 | authors: 4 | - name: "MMDetection Contributors" 5 | title: "OpenMMLab Detection Toolbox and Benchmark" 6 | date-released: 2018-08-22 7 | url: "https://github.com/open-mmlab/mmdetection" 8 | license: Apache-2.0 9 | -------------------------------------------------------------------------------- /software/mmdet/MANIFEST.in: -------------------------------------------------------------------------------- 1 | include requirements/*.txt 2 | include mmdet/VERSION 3 | include mmdet/.mim/model-index.yml 4 | include mmdet/.mim/demo/*/* 5 | recursive-include mmdet/.mim/configs *.py *.yml 6 | recursive-include mmdet/.mim/tools *.sh *.py 7 | -------------------------------------------------------------------------------- /software/mmdet/configs/atss/atss_r101_fpn_1x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './atss_r50_fpn_1x_coco.py' 2 | model = dict( 3 | backbone=dict( 4 | depth=101, 5 | init_cfg=dict(type='Pretrained', 6 | checkpoint='torchvision://resnet101'))) 7 | -------------------------------------------------------------------------------- /software/mmdet/configs/cascade_rcnn/cascade_mask_rcnn_r101_fpn_1x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './cascade_mask_rcnn_r50_fpn_1x_coco.py' 2 | model = dict( 3 | backbone=dict( 4 | depth=101, 5 | init_cfg=dict(type='Pretrained', 6 | checkpoint='torchvision://resnet101'))) 7 | -------------------------------------------------------------------------------- /software/mmdet/configs/cascade_rcnn/cascade_mask_rcnn_r101_fpn_20e_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './cascade_mask_rcnn_r50_fpn_20e_coco.py' 2 | model = dict( 3 | backbone=dict( 4 | depth=101, 5 | init_cfg=dict(type='Pretrained', 6 | checkpoint='torchvision://resnet101'))) 7 | -------------------------------------------------------------------------------- /software/mmdet/configs/cascade_rcnn/cascade_mask_rcnn_r101_fpn_mstrain_3x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './cascade_mask_rcnn_r50_fpn_mstrain_3x_coco.py' 2 | model = dict( 3 | backbone=dict( 4 | depth=101, 5 | init_cfg=dict(type='Pretrained', 6 | checkpoint='torchvision://resnet101'))) 7 | -------------------------------------------------------------------------------- /software/mmdet/configs/cascade_rcnn/cascade_mask_rcnn_r50_fpn_1x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/cascade_mask_rcnn_r50_fpn.py', 3 | '../_base_/datasets/coco_instance.py', 4 | '../_base_/schedules/schedule_1x.py', '../_base_/default_runtime.py' 5 | ] 6 | -------------------------------------------------------------------------------- /software/mmdet/configs/cascade_rcnn/cascade_mask_rcnn_r50_fpn_20e_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/cascade_mask_rcnn_r50_fpn.py', 3 | '../_base_/datasets/coco_instance.py', 4 | '../_base_/schedules/schedule_20e.py', '../_base_/default_runtime.py' 5 | ] 6 | -------------------------------------------------------------------------------- /software/mmdet/configs/cascade_rcnn/cascade_mask_rcnn_r50_fpn_mstrain_3x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../common/mstrain_3x_coco_instance.py', 3 | '../_base_/models/cascade_mask_rcnn_r50_fpn.py' 4 | ] 5 | -------------------------------------------------------------------------------- /software/mmdet/configs/cascade_rcnn/cascade_rcnn_r101_caffe_fpn_1x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './cascade_rcnn_r50_caffe_fpn_1x_coco.py' 2 | model = dict( 3 | backbone=dict( 4 | depth=101, 5 | init_cfg=dict( 6 | type='Pretrained', 7 | checkpoint='open-mmlab://detectron2/resnet101_caffe'))) 8 | -------------------------------------------------------------------------------- /software/mmdet/configs/cascade_rcnn/cascade_rcnn_r101_fpn_1x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './cascade_rcnn_r50_fpn_1x_coco.py' 2 | model = dict( 3 | backbone=dict( 4 | depth=101, 5 | init_cfg=dict(type='Pretrained', 6 | checkpoint='torchvision://resnet101'))) 7 | -------------------------------------------------------------------------------- /software/mmdet/configs/cascade_rcnn/cascade_rcnn_r101_fpn_20e_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './cascade_rcnn_r50_fpn_20e_coco.py' 2 | model = dict( 3 | backbone=dict( 4 | depth=101, 5 | init_cfg=dict(type='Pretrained', 6 | checkpoint='torchvision://resnet101'))) 7 | -------------------------------------------------------------------------------- /software/mmdet/configs/cascade_rcnn/cascade_rcnn_r50_fpn_1x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/cascade_rcnn_r50_fpn.py', 3 | '../_base_/datasets/coco_detection.py', 4 | '../_base_/schedules/schedule_1x.py', '../_base_/default_runtime.py' 5 | ] 6 | -------------------------------------------------------------------------------- /software/mmdet/configs/cascade_rcnn/cascade_rcnn_r50_fpn_20e_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './cascade_rcnn_r50_fpn_1x_coco.py' 2 | # learning policy 3 | lr_config = dict(step=[16, 19]) 4 | runner = dict(type='EpochBasedRunner', max_epochs=20) 5 | -------------------------------------------------------------------------------- /software/mmdet/configs/centernet/centernet_resnet18_140e_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './centernet_resnet18_dcnv2_140e_coco.py' 2 | 3 | model = dict(neck=dict(use_dcn=False)) 4 | -------------------------------------------------------------------------------- /software/mmdet/configs/dcn/cascade_mask_rcnn_r101_fpn_dconv_c3-c5_1x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = '../cascade_rcnn/cascade_mask_rcnn_r101_fpn_1x_coco.py' 2 | model = dict( 3 | backbone=dict( 4 | dcn=dict(type='DCN', deform_groups=1, fallback_on_stride=False), 5 | stage_with_dcn=(False, True, True, True))) 6 | -------------------------------------------------------------------------------- /software/mmdet/configs/dcn/cascade_mask_rcnn_r50_fpn_dconv_c3-c5_1x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = '../cascade_rcnn/cascade_mask_rcnn_r50_fpn_1x_coco.py' 2 | model = dict( 3 | backbone=dict( 4 | dcn=dict(type='DCN', deform_groups=1, fallback_on_stride=False), 5 | stage_with_dcn=(False, True, True, True))) 6 | -------------------------------------------------------------------------------- /software/mmdet/configs/dcn/cascade_mask_rcnn_x101_32x4d_fpn_dconv_c3-c5_1x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = '../cascade_rcnn/cascade_mask_rcnn_x101_32x4d_fpn_1x_coco.py' 2 | model = dict( 3 | backbone=dict( 4 | dcn=dict(type='DCN', deform_groups=1, fallback_on_stride=False), 5 | stage_with_dcn=(False, True, True, True))) 6 | -------------------------------------------------------------------------------- /software/mmdet/configs/dcn/cascade_rcnn_r101_fpn_dconv_c3-c5_1x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = '../cascade_rcnn/cascade_rcnn_r101_fpn_1x_coco.py' 2 | model = dict( 3 | backbone=dict( 4 | dcn=dict(type='DCN', deform_groups=1, fallback_on_stride=False), 5 | stage_with_dcn=(False, True, True, True))) 6 | -------------------------------------------------------------------------------- /software/mmdet/configs/dcn/cascade_rcnn_r50_fpn_dconv_c3-c5_1x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = '../cascade_rcnn/cascade_rcnn_r50_fpn_1x_coco.py' 2 | model = dict( 3 | backbone=dict( 4 | dcn=dict(type='DCN', deform_groups=1, fallback_on_stride=False), 5 | stage_with_dcn=(False, True, True, True))) 6 | -------------------------------------------------------------------------------- /software/mmdet/configs/dcn/faster_rcnn_r101_fpn_dconv_c3-c5_1x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = '../faster_rcnn/faster_rcnn_r101_fpn_1x_coco.py' 2 | model = dict( 3 | backbone=dict( 4 | dcn=dict(type='DCN', deform_groups=1, fallback_on_stride=False), 5 | stage_with_dcn=(False, True, True, True))) 6 | -------------------------------------------------------------------------------- /software/mmdet/configs/dcn/faster_rcnn_r50_fpn_dconv_c3-c5_1x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = '../faster_rcnn/faster_rcnn_r50_fpn_1x_coco.py' 2 | model = dict( 3 | backbone=dict( 4 | dcn=dict(type='DCN', deform_groups=1, fallback_on_stride=False), 5 | stage_with_dcn=(False, True, True, True))) 6 | -------------------------------------------------------------------------------- /software/mmdet/configs/dcn/mask_rcnn_r101_fpn_dconv_c3-c5_1x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = '../mask_rcnn/mask_rcnn_r101_fpn_1x_coco.py' 2 | model = dict( 3 | backbone=dict( 4 | dcn=dict(type='DCN', deform_groups=1, fallback_on_stride=False), 5 | stage_with_dcn=(False, True, True, True))) 6 | -------------------------------------------------------------------------------- /software/mmdet/configs/dcn/mask_rcnn_r50_fpn_dconv_c3-c5_1x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = '../mask_rcnn/mask_rcnn_r50_fpn_1x_coco.py' 2 | model = dict( 3 | backbone=dict( 4 | dcn=dict(type='DCN', deform_groups=1, fallback_on_stride=False), 5 | stage_with_dcn=(False, True, True, True))) 6 | -------------------------------------------------------------------------------- /software/mmdet/configs/dcnv2/faster_rcnn_r50_fpn_mdconv_c3-c5_1x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = '../faster_rcnn/faster_rcnn_r50_fpn_1x_coco.py' 2 | model = dict( 3 | backbone=dict( 4 | dcn=dict(type='DCNv2', deform_groups=1, fallback_on_stride=False), 5 | stage_with_dcn=(False, True, True, True))) 6 | -------------------------------------------------------------------------------- /software/mmdet/configs/dcnv2/faster_rcnn_r50_fpn_mdconv_c3-c5_group4_1x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = '../faster_rcnn/faster_rcnn_r50_fpn_1x_coco.py' 2 | model = dict( 3 | backbone=dict( 4 | dcn=dict(type='DCNv2', deform_groups=4, fallback_on_stride=False), 5 | stage_with_dcn=(False, True, True, True))) 6 | -------------------------------------------------------------------------------- /software/mmdet/configs/dcnv2/mask_rcnn_r50_fpn_mdconv_c3-c5_1x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = '../mask_rcnn/mask_rcnn_r50_fpn_1x_coco.py' 2 | model = dict( 3 | backbone=dict( 4 | dcn=dict(type='DCNv2', deform_groups=1, fallback_on_stride=False), 5 | stage_with_dcn=(False, True, True, True))) 6 | -------------------------------------------------------------------------------- /software/mmdet/configs/deformable_detr/deformable_detr_refine_r50_16x2_50e_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = 'deformable_detr_r50_16x2_50e_coco.py' 2 | model = dict(bbox_head=dict(with_box_refine=True)) 3 | -------------------------------------------------------------------------------- /software/mmdet/configs/deformable_detr/deformable_detr_twostage_refine_r50_16x2_50e_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = 'deformable_detr_refine_r50_16x2_50e_coco.py' 2 | model = dict(bbox_head=dict(as_two_stage=True)) 3 | -------------------------------------------------------------------------------- /software/mmdet/configs/detectors/htc_r50_sac_1x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = '../htc/htc_r50_fpn_1x_coco.py' 2 | 3 | model = dict( 4 | backbone=dict( 5 | type='DetectoRS_ResNet', 6 | conv_cfg=dict(type='ConvAWS'), 7 | sac=dict(type='SAC', use_deform=True), 8 | stage_with_sac=(False, True, True, True))) 9 | -------------------------------------------------------------------------------- /software/mmdet/configs/fast_rcnn/fast_rcnn_r101_caffe_fpn_1x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './fast_rcnn_r50_caffe_fpn_1x_coco.py' 2 | model = dict( 3 | backbone=dict( 4 | depth=101, 5 | init_cfg=dict( 6 | type='Pretrained', 7 | checkpoint='open-mmlab://detectron2/resnet101_caffe'))) 8 | -------------------------------------------------------------------------------- /software/mmdet/configs/fast_rcnn/fast_rcnn_r101_fpn_1x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './fast_rcnn_r50_fpn_1x_coco.py' 2 | model = dict( 3 | backbone=dict( 4 | depth=101, 5 | init_cfg=dict(type='Pretrained', 6 | checkpoint='torchvision://resnet101'))) 7 | -------------------------------------------------------------------------------- /software/mmdet/configs/fast_rcnn/fast_rcnn_r101_fpn_2x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './fast_rcnn_r50_fpn_2x_coco.py' 2 | model = dict( 3 | backbone=dict( 4 | depth=101, 5 | init_cfg=dict(type='Pretrained', 6 | checkpoint='torchvision://resnet101'))) 7 | -------------------------------------------------------------------------------- /software/mmdet/configs/fast_rcnn/fast_rcnn_r50_fpn_2x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './fast_rcnn_r50_fpn_1x_coco.py' 2 | 3 | # learning policy 4 | lr_config = dict(step=[16, 22]) 5 | runner = dict(type='EpochBasedRunner', max_epochs=24) 6 | -------------------------------------------------------------------------------- /software/mmdet/configs/faster_rcnn/faster_rcnn_r101_caffe_fpn_1x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './faster_rcnn_r50_caffe_fpn_1x_coco.py' 2 | model = dict( 3 | backbone=dict( 4 | depth=101, 5 | init_cfg=dict( 6 | type='Pretrained', 7 | checkpoint='open-mmlab://detectron2/resnet101_caffe'))) 8 | -------------------------------------------------------------------------------- /software/mmdet/configs/faster_rcnn/faster_rcnn_r101_fpn_1x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './faster_rcnn_r50_fpn_1x_coco.py' 2 | model = dict( 3 | backbone=dict( 4 | depth=101, 5 | init_cfg=dict(type='Pretrained', 6 | checkpoint='torchvision://resnet101'))) 7 | -------------------------------------------------------------------------------- /software/mmdet/configs/faster_rcnn/faster_rcnn_r101_fpn_2x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './faster_rcnn_r50_fpn_2x_coco.py' 2 | model = dict( 3 | backbone=dict( 4 | depth=101, 5 | init_cfg=dict(type='Pretrained', 6 | checkpoint='torchvision://resnet101'))) 7 | -------------------------------------------------------------------------------- /software/mmdet/configs/faster_rcnn/faster_rcnn_r101_fpn_mstrain_3x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = 'faster_rcnn_r50_fpn_mstrain_3x_coco.py' 2 | 3 | model = dict( 4 | backbone=dict( 5 | depth=101, 6 | init_cfg=dict(type='Pretrained', 7 | checkpoint='torchvision://resnet101'))) 8 | -------------------------------------------------------------------------------- /software/mmdet/configs/faster_rcnn/faster_rcnn_r50_caffe_dc5_mstrain_3x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './faster_rcnn_r50_caffe_dc5_mstrain_1x_coco.py' 2 | # learning policy 3 | lr_config = dict(step=[28, 34]) 4 | runner = dict(type='EpochBasedRunner', max_epochs=36) 5 | -------------------------------------------------------------------------------- /software/mmdet/configs/faster_rcnn/faster_rcnn_r50_caffe_fpn_mstrain_2x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './faster_rcnn_r50_caffe_fpn_mstrain_1x_coco.py' 2 | # learning policy 3 | lr_config = dict(step=[16, 23]) 4 | runner = dict(type='EpochBasedRunner', max_epochs=24) 5 | -------------------------------------------------------------------------------- /software/mmdet/configs/faster_rcnn/faster_rcnn_r50_fpn_1x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/faster_rcnn_r50_fpn.py', 3 | '../_base_/datasets/coco_detection.py', 4 | '../_base_/schedules/schedule_1x.py', '../_base_/default_runtime.py' 5 | ] 6 | -------------------------------------------------------------------------------- /software/mmdet/configs/faster_rcnn/faster_rcnn_r50_fpn_2x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/faster_rcnn_r50_fpn.py', 3 | '../_base_/datasets/coco_detection.py', 4 | '../_base_/schedules/schedule_2x.py', '../_base_/default_runtime.py' 5 | ] 6 | -------------------------------------------------------------------------------- /software/mmdet/configs/faster_rcnn/faster_rcnn_r50_fpn_bounded_iou_1x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './faster_rcnn_r50_fpn_1x_coco.py' 2 | model = dict( 3 | roi_head=dict( 4 | bbox_head=dict( 5 | reg_decoded_bbox=True, 6 | loss_bbox=dict(type='BoundedIoULoss', loss_weight=10.0)))) 7 | -------------------------------------------------------------------------------- /software/mmdet/configs/faster_rcnn/faster_rcnn_r50_fpn_ciou_1x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './faster_rcnn_r50_fpn_1x_coco.py' 2 | model = dict( 3 | roi_head=dict( 4 | bbox_head=dict( 5 | reg_decoded_bbox=True, 6 | loss_bbox=dict(type='CIoULoss', loss_weight=12.0)))) 7 | -------------------------------------------------------------------------------- /software/mmdet/configs/faster_rcnn/faster_rcnn_r50_fpn_fp16_1x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './faster_rcnn_r50_fpn_1x_coco.py' 2 | # fp16 settings 3 | fp16 = dict(loss_scale=512.) 4 | -------------------------------------------------------------------------------- /software/mmdet/configs/faster_rcnn/faster_rcnn_r50_fpn_giou_1x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './faster_rcnn_r50_fpn_1x_coco.py' 2 | model = dict( 3 | roi_head=dict( 4 | bbox_head=dict( 5 | reg_decoded_bbox=True, 6 | loss_bbox=dict(type='GIoULoss', loss_weight=10.0)))) 7 | -------------------------------------------------------------------------------- /software/mmdet/configs/faster_rcnn/faster_rcnn_r50_fpn_iou_1x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './faster_rcnn_r50_fpn_1x_coco.py' 2 | model = dict( 3 | roi_head=dict( 4 | bbox_head=dict( 5 | reg_decoded_bbox=True, 6 | loss_bbox=dict(type='IoULoss', loss_weight=10.0)))) 7 | -------------------------------------------------------------------------------- /software/mmdet/configs/faster_rcnn/faster_rcnn_r50_fpn_mstrain_3x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../common/mstrain_3x_coco.py', '../_base_/models/faster_rcnn_r50_fpn.py' 3 | ] 4 | -------------------------------------------------------------------------------- /software/mmdet/configs/faster_rcnn/faster_rcnn_r50_fpn_ohem_1x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './faster_rcnn_r50_fpn_1x_coco.py' 2 | model = dict(train_cfg=dict(rcnn=dict(sampler=dict(type='OHEMSampler')))) 3 | -------------------------------------------------------------------------------- /software/mmdet/configs/fcos/fcos_center_r50_caffe_fpn_gn-head_1x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './fcos_r50_caffe_fpn_gn-head_1x_coco.py' 2 | model = dict(bbox_head=dict(center_sampling=True, center_sample_radius=1.5)) 3 | -------------------------------------------------------------------------------- /software/mmdet/configs/fcos/fcos_r101_caffe_fpn_gn-head_1x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './fcos_r50_caffe_fpn_gn-head_1x_coco.py' 2 | model = dict( 3 | backbone=dict( 4 | depth=101, 5 | init_cfg=dict( 6 | type='Pretrained', 7 | checkpoint='open-mmlab://detectron/resnet101_caffe'))) 8 | -------------------------------------------------------------------------------- /software/mmdet/configs/fcos/fcos_r50_caffe_fpn_gn-head_4x4_1x_coco.py: -------------------------------------------------------------------------------- 1 | # TODO: Remove this config after benchmarking all related configs 2 | _base_ = 'fcos_r50_caffe_fpn_gn-head_1x_coco.py' 3 | 4 | data = dict(samples_per_gpu=4, workers_per_gpu=4) 5 | -------------------------------------------------------------------------------- /software/mmdet/configs/foveabox/fovea_r101_fpn_4x4_1x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './fovea_r50_fpn_4x4_1x_coco.py' 2 | model = dict( 3 | backbone=dict( 4 | depth=101, 5 | init_cfg=dict(type='Pretrained', 6 | checkpoint='torchvision://resnet101'))) 7 | -------------------------------------------------------------------------------- /software/mmdet/configs/foveabox/fovea_r101_fpn_4x4_2x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './fovea_r50_fpn_4x4_2x_coco.py' 2 | model = dict( 3 | backbone=dict( 4 | depth=101, 5 | init_cfg=dict(type='Pretrained', 6 | checkpoint='torchvision://resnet101'))) 7 | -------------------------------------------------------------------------------- /software/mmdet/configs/foveabox/fovea_r50_fpn_4x4_2x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './fovea_r50_fpn_4x4_1x_coco.py' 2 | # learning policy 3 | lr_config = dict(step=[16, 22]) 4 | runner = dict(type='EpochBasedRunner', max_epochs=24) 5 | -------------------------------------------------------------------------------- /software/mmdet/configs/fpg/retinanet_r50_fpg-chn128_crop640_50e_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = 'retinanet_r50_fpg_crop640_50e_coco.py' 2 | 3 | model = dict( 4 | neck=dict(out_channels=128, inter_channels=128), 5 | bbox_head=dict(in_channels=128)) 6 | -------------------------------------------------------------------------------- /software/mmdet/configs/free_anchor/retinanet_free_anchor_r101_fpn_1x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './retinanet_free_anchor_r50_fpn_1x_coco.py' 2 | model = dict( 3 | backbone=dict( 4 | depth=101, 5 | init_cfg=dict(type='Pretrained', 6 | checkpoint='torchvision://resnet101'))) 7 | -------------------------------------------------------------------------------- /software/mmdet/configs/fsaf/fsaf_r101_fpn_1x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './fsaf_r50_fpn_1x_coco.py' 2 | model = dict( 3 | backbone=dict( 4 | depth=101, 5 | init_cfg=dict(type='Pretrained', 6 | checkpoint='torchvision://resnet101'))) 7 | -------------------------------------------------------------------------------- /software/mmdet/configs/gcnet/cascade_mask_rcnn_x101_32x4d_fpn_syncbn-backbone_1x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = '../cascade_rcnn/cascade_mask_rcnn_x101_32x4d_fpn_1x_coco.py' 2 | model = dict( 3 | backbone=dict( 4 | norm_cfg=dict(type='SyncBN', requires_grad=True), norm_eval=False)) 5 | -------------------------------------------------------------------------------- /software/mmdet/configs/gcnet/cascade_mask_rcnn_x101_32x4d_fpn_syncbn-backbone_dconv_c3-c5_1x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = '../dcn/cascade_mask_rcnn_x101_32x4d_fpn_dconv_c3-c5_1x_coco.py' 2 | model = dict( 3 | backbone=dict( 4 | norm_cfg=dict(type='SyncBN', requires_grad=True), norm_eval=False)) 5 | -------------------------------------------------------------------------------- /software/mmdet/configs/gcnet/mask_rcnn_r101_fpn_syncbn-backbone_1x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = '../mask_rcnn/mask_rcnn_r101_fpn_1x_coco.py' 2 | model = dict( 3 | backbone=dict( 4 | norm_cfg=dict(type='SyncBN', requires_grad=True), norm_eval=False)) 5 | -------------------------------------------------------------------------------- /software/mmdet/configs/gcnet/mask_rcnn_r50_fpn_syncbn-backbone_1x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = '../mask_rcnn/mask_rcnn_r50_fpn_1x_coco.py' 2 | model = dict( 3 | backbone=dict( 4 | norm_cfg=dict(type='SyncBN', requires_grad=True), norm_eval=False)) 5 | -------------------------------------------------------------------------------- /software/mmdet/configs/gcnet/mask_rcnn_x101_32x4d_fpn_syncbn-backbone_1x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = '../mask_rcnn/mask_rcnn_x101_32x4d_fpn_1x_coco.py' 2 | model = dict( 3 | backbone=dict( 4 | norm_cfg=dict(type='SyncBN', requires_grad=True), norm_eval=False)) 5 | -------------------------------------------------------------------------------- /software/mmdet/configs/ghm/retinanet_ghm_r101_fpn_1x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './retinanet_ghm_r50_fpn_1x_coco.py' 2 | model = dict( 3 | backbone=dict( 4 | depth=101, 5 | init_cfg=dict(type='Pretrained', 6 | checkpoint='torchvision://resnet101'))) 7 | -------------------------------------------------------------------------------- /software/mmdet/configs/gn+ws/faster_rcnn_r101_fpn_gn_ws-all_1x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './faster_rcnn_r50_fpn_gn_ws-all_1x_coco.py' 2 | model = dict( 3 | backbone=dict( 4 | depth=101, 5 | init_cfg=dict( 6 | type='Pretrained', checkpoint='open-mmlab://jhu/resnet101_gn_ws'))) 7 | -------------------------------------------------------------------------------- /software/mmdet/configs/gn+ws/mask_rcnn_r101_fpn_gn_ws-all_20_23_24e_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './mask_rcnn_r101_fpn_gn_ws-all_2x_coco.py' 2 | # learning policy 3 | lr_config = dict(step=[20, 23]) 4 | runner = dict(type='EpochBasedRunner', max_epochs=24) 5 | -------------------------------------------------------------------------------- /software/mmdet/configs/gn+ws/mask_rcnn_r101_fpn_gn_ws-all_2x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './mask_rcnn_r50_fpn_gn_ws-all_2x_coco.py' 2 | model = dict( 3 | backbone=dict( 4 | depth=101, 5 | init_cfg=dict( 6 | type='Pretrained', checkpoint='open-mmlab://jhu/resnet101_gn_ws'))) 7 | -------------------------------------------------------------------------------- /software/mmdet/configs/gn+ws/mask_rcnn_r50_fpn_gn_ws-all_20_23_24e_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './mask_rcnn_r50_fpn_gn_ws-all_2x_coco.py' 2 | # learning policy 3 | lr_config = dict(step=[20, 23]) 4 | runner = dict(type='EpochBasedRunner', max_epochs=24) 5 | -------------------------------------------------------------------------------- /software/mmdet/configs/gn+ws/mask_rcnn_x101_32x4d_fpn_gn_ws-all_20_23_24e_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './mask_rcnn_x101_32x4d_fpn_gn_ws-all_2x_coco.py' 2 | # learning policy 3 | lr_config = dict(step=[20, 23]) 4 | runner = dict(type='EpochBasedRunner', max_epochs=24) 5 | -------------------------------------------------------------------------------- /software/mmdet/configs/gn+ws/mask_rcnn_x50_32x4d_fpn_gn_ws-all_20_23_24e_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './mask_rcnn_x50_32x4d_fpn_gn_ws-all_2x_coco.py' 2 | # learning policy 3 | lr_config = dict(step=[20, 23]) 4 | runner = dict(type='EpochBasedRunner', max_epochs=24) 5 | -------------------------------------------------------------------------------- /software/mmdet/configs/gn/mask_rcnn_r101_fpn_gn-all_2x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './mask_rcnn_r50_fpn_gn-all_2x_coco.py' 2 | model = dict( 3 | backbone=dict( 4 | depth=101, 5 | init_cfg=dict( 6 | type='Pretrained', 7 | checkpoint='open-mmlab://detectron/resnet101_gn'))) 8 | -------------------------------------------------------------------------------- /software/mmdet/configs/gn/mask_rcnn_r101_fpn_gn-all_3x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './mask_rcnn_r101_fpn_gn-all_2x_coco.py' 2 | 3 | # learning policy 4 | lr_config = dict(step=[28, 34]) 5 | runner = dict(type='EpochBasedRunner', max_epochs=36) 6 | -------------------------------------------------------------------------------- /software/mmdet/configs/gn/mask_rcnn_r50_fpn_gn-all_3x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './mask_rcnn_r50_fpn_gn-all_2x_coco.py' 2 | 3 | # learning policy 4 | lr_config = dict(step=[28, 34]) 5 | runner = dict(type='EpochBasedRunner', max_epochs=36) 6 | -------------------------------------------------------------------------------- /software/mmdet/configs/gn/mask_rcnn_r50_fpn_gn-all_contrib_3x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './mask_rcnn_r50_fpn_gn-all_contrib_2x_coco.py' 2 | 3 | # learning policy 4 | lr_config = dict(step=[28, 34]) 5 | runner = dict(type='EpochBasedRunner', max_epochs=36) 6 | -------------------------------------------------------------------------------- /software/mmdet/configs/grid_rcnn/grid_rcnn_r101_fpn_gn-head_2x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './grid_rcnn_r50_fpn_gn-head_2x_coco.py' 2 | 3 | model = dict( 4 | backbone=dict( 5 | depth=101, 6 | init_cfg=dict(type='Pretrained', 7 | checkpoint='torchvision://resnet101'))) 8 | -------------------------------------------------------------------------------- /software/mmdet/configs/guided_anchoring/ga_faster_r101_caffe_fpn_1x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './ga_faster_r50_caffe_fpn_1x_coco.py' 2 | model = dict( 3 | backbone=dict( 4 | depth=101, 5 | init_cfg=dict( 6 | type='Pretrained', 7 | checkpoint='open-mmlab://detectron2/resnet101_caffe'))) 8 | -------------------------------------------------------------------------------- /software/mmdet/configs/guided_anchoring/ga_retinanet_r101_caffe_fpn_1x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './ga_retinanet_r50_caffe_fpn_1x_coco.py' 2 | model = dict( 3 | backbone=dict( 4 | depth=101, 5 | init_cfg=dict( 6 | type='Pretrained', 7 | checkpoint='open-mmlab://detectron2/resnet101_caffe'))) 8 | -------------------------------------------------------------------------------- /software/mmdet/configs/hrnet/faster_rcnn_hrnetv2p_w18_2x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './faster_rcnn_hrnetv2p_w18_1x_coco.py' 2 | 3 | # learning policy 4 | lr_config = dict(step=[16, 22]) 5 | runner = dict(type='EpochBasedRunner', max_epochs=24) 6 | -------------------------------------------------------------------------------- /software/mmdet/configs/hrnet/faster_rcnn_hrnetv2p_w32_2x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './faster_rcnn_hrnetv2p_w32_1x_coco.py' 2 | # learning policy 3 | lr_config = dict(step=[16, 22]) 4 | runner = dict(type='EpochBasedRunner', max_epochs=24) 5 | -------------------------------------------------------------------------------- /software/mmdet/configs/hrnet/faster_rcnn_hrnetv2p_w40_2x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './faster_rcnn_hrnetv2p_w40_1x_coco.py' 2 | # learning policy 3 | lr_config = dict(step=[16, 22]) 4 | runner = dict(type='EpochBasedRunner', max_epochs=24) 5 | -------------------------------------------------------------------------------- /software/mmdet/configs/hrnet/fcos_hrnetv2p_w18_gn-head_4x4_2x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './fcos_hrnetv2p_w18_gn-head_4x4_1x_coco.py' 2 | # learning policy 3 | lr_config = dict(step=[16, 22]) 4 | runner = dict(type='EpochBasedRunner', max_epochs=24) 5 | -------------------------------------------------------------------------------- /software/mmdet/configs/hrnet/fcos_hrnetv2p_w32_gn-head_4x4_2x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './fcos_hrnetv2p_w32_gn-head_4x4_1x_coco.py' 2 | # learning policy 3 | lr_config = dict(step=[16, 22]) 4 | runner = dict(type='EpochBasedRunner', max_epochs=24) 5 | -------------------------------------------------------------------------------- /software/mmdet/configs/hrnet/htc_hrnetv2p_w40_28e_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './htc_hrnetv2p_w40_20e_coco.py' 2 | # learning policy 3 | lr_config = dict(step=[24, 27]) 4 | runner = dict(type='EpochBasedRunner', max_epochs=28) 5 | -------------------------------------------------------------------------------- /software/mmdet/configs/hrnet/htc_x101_64x4d_fpn_16x1_28e_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = '../htc/htc_x101_64x4d_fpn_16x1_20e_coco.py' 2 | # learning policy 3 | lr_config = dict(step=[24, 27]) 4 | runner = dict(type='EpochBasedRunner', max_epochs=28) 5 | -------------------------------------------------------------------------------- /software/mmdet/configs/hrnet/mask_rcnn_hrnetv2p_w18_2x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './mask_rcnn_hrnetv2p_w18_1x_coco.py' 2 | # learning policy 3 | lr_config = dict(step=[16, 22]) 4 | runner = dict(type='EpochBasedRunner', max_epochs=24) 5 | -------------------------------------------------------------------------------- /software/mmdet/configs/hrnet/mask_rcnn_hrnetv2p_w32_2x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './mask_rcnn_hrnetv2p_w32_1x_coco.py' 2 | # learning policy 3 | lr_config = dict(step=[16, 22]) 4 | runner = dict(type='EpochBasedRunner', max_epochs=24) 5 | -------------------------------------------------------------------------------- /software/mmdet/configs/hrnet/mask_rcnn_hrnetv2p_w40_2x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './mask_rcnn_hrnetv2p_w40_1x_coco.py' 2 | # learning policy 3 | lr_config = dict(step=[16, 22]) 4 | runner = dict(type='EpochBasedRunner', max_epochs=24) 5 | -------------------------------------------------------------------------------- /software/mmdet/configs/htc/htc_r50_fpn_20e_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './htc_r50_fpn_1x_coco.py' 2 | # learning policy 3 | lr_config = dict(step=[16, 19]) 4 | runner = dict(type='EpochBasedRunner', max_epochs=20) 5 | -------------------------------------------------------------------------------- /software/mmdet/configs/instaboost/cascade_mask_rcnn_r101_fpn_instaboost_4x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './cascade_mask_rcnn_r50_fpn_instaboost_4x_coco.py' 2 | 3 | model = dict( 4 | backbone=dict( 5 | depth=101, 6 | init_cfg=dict(type='Pretrained', 7 | checkpoint='torchvision://resnet101'))) 8 | -------------------------------------------------------------------------------- /software/mmdet/configs/instaboost/mask_rcnn_r101_fpn_instaboost_4x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './mask_rcnn_r50_fpn_instaboost_4x_coco.py' 2 | model = dict( 3 | backbone=dict( 4 | depth=101, 5 | init_cfg=dict(type='Pretrained', 6 | checkpoint='torchvision://resnet101'))) 7 | -------------------------------------------------------------------------------- /software/mmdet/configs/libra_rcnn/libra_faster_rcnn_r101_fpn_1x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './libra_faster_rcnn_r50_fpn_1x_coco.py' 2 | model = dict( 3 | backbone=dict( 4 | depth=101, 5 | init_cfg=dict(type='Pretrained', 6 | checkpoint='torchvision://resnet101'))) 7 | -------------------------------------------------------------------------------- /software/mmdet/configs/lvis/mask_rcnn_r101_fpn_sample1e-3_mstrain_1x_lvis_v1.py: -------------------------------------------------------------------------------- 1 | _base_ = './mask_rcnn_r50_fpn_sample1e-3_mstrain_1x_lvis_v1.py' 2 | model = dict( 3 | backbone=dict( 4 | depth=101, 5 | init_cfg=dict(type='Pretrained', 6 | checkpoint='torchvision://resnet101'))) 7 | -------------------------------------------------------------------------------- /software/mmdet/configs/lvis/mask_rcnn_r101_fpn_sample1e-3_mstrain_2x_lvis_v0.5.py: -------------------------------------------------------------------------------- 1 | _base_ = './mask_rcnn_r50_fpn_sample1e-3_mstrain_2x_lvis_v0.5.py' 2 | model = dict( 3 | backbone=dict( 4 | depth=101, 5 | init_cfg=dict(type='Pretrained', 6 | checkpoint='torchvision://resnet101'))) 7 | -------------------------------------------------------------------------------- /software/mmdet/configs/mask2former/mask2former_r101_lsj_8x2_50e_coco-panoptic.py: -------------------------------------------------------------------------------- 1 | _base_ = './mask2former_r50_lsj_8x2_50e_coco-panoptic.py' 2 | 3 | model = dict( 4 | backbone=dict( 5 | depth=101, 6 | init_cfg=dict(type='Pretrained', 7 | checkpoint='torchvision://resnet101'))) 8 | -------------------------------------------------------------------------------- /software/mmdet/configs/mask2former/mask2former_r101_lsj_8x2_50e_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = ['./mask2former_r50_lsj_8x2_50e_coco.py'] 2 | 3 | model = dict( 4 | backbone=dict( 5 | depth=101, 6 | init_cfg=dict(type='Pretrained', 7 | checkpoint='torchvision://resnet101'))) 8 | -------------------------------------------------------------------------------- /software/mmdet/configs/mask_rcnn/mask_rcnn_r101_caffe_fpn_1x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './mask_rcnn_r50_caffe_fpn_1x_coco.py' 2 | model = dict( 3 | backbone=dict( 4 | depth=101, 5 | init_cfg=dict( 6 | type='Pretrained', 7 | checkpoint='open-mmlab://detectron2/resnet101_caffe'))) 8 | -------------------------------------------------------------------------------- /software/mmdet/configs/mask_rcnn/mask_rcnn_r101_fpn_1x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './mask_rcnn_r50_fpn_1x_coco.py' 2 | model = dict( 3 | backbone=dict( 4 | depth=101, 5 | init_cfg=dict(type='Pretrained', 6 | checkpoint='torchvision://resnet101'))) 7 | -------------------------------------------------------------------------------- /software/mmdet/configs/mask_rcnn/mask_rcnn_r101_fpn_2x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './mask_rcnn_r50_fpn_2x_coco.py' 2 | model = dict( 3 | backbone=dict( 4 | depth=101, 5 | init_cfg=dict(type='Pretrained', 6 | checkpoint='torchvision://resnet101'))) 7 | -------------------------------------------------------------------------------- /software/mmdet/configs/mask_rcnn/mask_rcnn_r50_caffe_fpn_mstrain-poly_2x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './mask_rcnn_r50_caffe_fpn_mstrain-poly_1x_coco.py' 2 | # learning policy 3 | lr_config = dict(step=[16, 23]) 4 | runner = dict(type='EpochBasedRunner', max_epochs=24) 5 | -------------------------------------------------------------------------------- /software/mmdet/configs/mask_rcnn/mask_rcnn_r50_caffe_fpn_mstrain-poly_3x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './mask_rcnn_r50_caffe_fpn_mstrain-poly_1x_coco.py' 2 | # learning policy 3 | lr_config = dict(step=[28, 34]) 4 | runner = dict(type='EpochBasedRunner', max_epochs=36) 5 | -------------------------------------------------------------------------------- /software/mmdet/configs/mask_rcnn/mask_rcnn_r50_fpn_1x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/mask_rcnn_r50_fpn.py', 3 | '../_base_/datasets/coco_instance.py', 4 | '../_base_/schedules/schedule_1x.py', '../_base_/default_runtime.py' 5 | ] 6 | -------------------------------------------------------------------------------- /software/mmdet/configs/mask_rcnn/mask_rcnn_r50_fpn_2x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/mask_rcnn_r50_fpn.py', 3 | '../_base_/datasets/coco_instance.py', 4 | '../_base_/schedules/schedule_2x.py', '../_base_/default_runtime.py' 5 | ] 6 | -------------------------------------------------------------------------------- /software/mmdet/configs/mask_rcnn/mask_rcnn_r50_fpn_fp16_1x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './mask_rcnn_r50_fpn_1x_coco.py' 2 | # fp16 settings 3 | fp16 = dict(loss_scale=512.) 4 | -------------------------------------------------------------------------------- /software/mmdet/configs/mask_rcnn/mask_rcnn_r50_fpn_mstrain-poly_3x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../common/mstrain-poly_3x_coco_instance.py', 3 | '../_base_/models/mask_rcnn_r50_fpn.py' 4 | ] 5 | -------------------------------------------------------------------------------- /software/mmdet/configs/ms_rcnn/ms_rcnn_r101_caffe_fpn_1x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './ms_rcnn_r50_caffe_fpn_1x_coco.py' 2 | model = dict( 3 | backbone=dict( 4 | depth=101, 5 | init_cfg=dict( 6 | type='Pretrained', 7 | checkpoint='open-mmlab://detectron2/resnet101_caffe'))) 8 | -------------------------------------------------------------------------------- /software/mmdet/configs/ms_rcnn/ms_rcnn_r101_caffe_fpn_2x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './ms_rcnn_r101_caffe_fpn_1x_coco.py' 2 | # learning policy 3 | lr_config = dict(step=[16, 22]) 4 | runner = dict(type='EpochBasedRunner', max_epochs=24) 5 | -------------------------------------------------------------------------------- /software/mmdet/configs/ms_rcnn/ms_rcnn_r50_caffe_fpn_2x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './ms_rcnn_r50_caffe_fpn_1x_coco.py' 2 | # learning policy 3 | lr_config = dict(step=[16, 22]) 4 | runner = dict(type='EpochBasedRunner', max_epochs=24) 5 | -------------------------------------------------------------------------------- /software/mmdet/configs/ms_rcnn/ms_rcnn_x101_64x4d_fpn_2x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './ms_rcnn_x101_64x4d_fpn_1x_coco.py' 2 | # learning policy 3 | lr_config = dict(step=[16, 22]) 4 | runner = dict(type='EpochBasedRunner', max_epochs=24) 5 | -------------------------------------------------------------------------------- /software/mmdet/configs/openimages/faster_rcnn_r50_fpn_32x2_cas_1x_openimages.py: -------------------------------------------------------------------------------- 1 | _base_ = ['faster_rcnn_r50_fpn_32x2_1x_openimages.py'] 2 | 3 | # Use ClassAwareSampler 4 | data = dict( 5 | train_dataloader=dict(class_aware_sampler=dict(num_sample_class=1))) 6 | -------------------------------------------------------------------------------- /software/mmdet/configs/openimages/faster_rcnn_r50_fpn_32x2_cas_1x_openimages_challenge.py: -------------------------------------------------------------------------------- 1 | _base_ = ['faster_rcnn_r50_fpn_32x2_1x_openimages_challenge.py'] 2 | 3 | # Use ClassAwareSampler 4 | data = dict( 5 | train_dataloader=dict(class_aware_sampler=dict(num_sample_class=1))) 6 | -------------------------------------------------------------------------------- /software/mmdet/configs/paa/paa_r101_fpn_1x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './paa_r50_fpn_1x_coco.py' 2 | model = dict( 3 | backbone=dict( 4 | depth=101, 5 | init_cfg=dict(type='Pretrained', 6 | checkpoint='torchvision://resnet101'))) 7 | -------------------------------------------------------------------------------- /software/mmdet/configs/paa/paa_r101_fpn_2x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './paa_r101_fpn_1x_coco.py' 2 | lr_config = dict(step=[16, 22]) 3 | runner = dict(type='EpochBasedRunner', max_epochs=24) 4 | -------------------------------------------------------------------------------- /software/mmdet/configs/paa/paa_r101_fpn_mstrain_3x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './paa_r50_fpn_mstrain_3x_coco.py' 2 | model = dict( 3 | backbone=dict( 4 | depth=101, 5 | init_cfg=dict(type='Pretrained', 6 | checkpoint='torchvision://resnet101'))) 7 | -------------------------------------------------------------------------------- /software/mmdet/configs/paa/paa_r50_fpn_1.5x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './paa_r50_fpn_1x_coco.py' 2 | lr_config = dict(step=[12, 16]) 3 | runner = dict(type='EpochBasedRunner', max_epochs=18) 4 | -------------------------------------------------------------------------------- /software/mmdet/configs/paa/paa_r50_fpn_2x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './paa_r50_fpn_1x_coco.py' 2 | lr_config = dict(step=[16, 22]) 3 | runner = dict(type='EpochBasedRunner', max_epochs=24) 4 | -------------------------------------------------------------------------------- /software/mmdet/configs/pafpn/faster_rcnn_r50_pafpn_1x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = '../faster_rcnn/faster_rcnn_r50_fpn_1x_coco.py' 2 | 3 | model = dict( 4 | neck=dict( 5 | type='PAFPN', 6 | in_channels=[256, 512, 1024, 2048], 7 | out_channels=256, 8 | num_outs=5)) 9 | -------------------------------------------------------------------------------- /software/mmdet/configs/panoptic_fpn/panoptic_fpn_r101_fpn_1x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './panoptic_fpn_r50_fpn_1x_coco.py' 2 | model = dict( 3 | backbone=dict( 4 | depth=101, 5 | init_cfg=dict(type='Pretrained', 6 | checkpoint='torchvision://resnet101'))) 7 | -------------------------------------------------------------------------------- /software/mmdet/configs/panoptic_fpn/panoptic_fpn_r101_fpn_mstrain_3x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './panoptic_fpn_r50_fpn_mstrain_3x_coco.py' 2 | model = dict( 3 | backbone=dict( 4 | depth=101, 5 | init_cfg=dict(type='Pretrained', 6 | checkpoint='torchvision://resnet101'))) 7 | -------------------------------------------------------------------------------- /software/mmdet/configs/pisa/pisa_ssd300_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = '../ssd/ssd300_coco.py' 2 | 3 | model = dict( 4 | bbox_head=dict(type='PISASSDHead'), 5 | train_cfg=dict(isr=dict(k=2., bias=0.), carl=dict(k=1., bias=0.2))) 6 | 7 | optimizer_config = dict( 8 | _delete_=True, grad_clip=dict(max_norm=35, norm_type=2)) 9 | -------------------------------------------------------------------------------- /software/mmdet/configs/pisa/pisa_ssd512_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = '../ssd/ssd512_coco.py' 2 | 3 | model = dict( 4 | bbox_head=dict(type='PISASSDHead'), 5 | train_cfg=dict(isr=dict(k=2., bias=0.), carl=dict(k=1., bias=0.2))) 6 | 7 | optimizer_config = dict( 8 | _delete_=True, grad_clip=dict(max_norm=35, norm_type=2)) 9 | -------------------------------------------------------------------------------- /software/mmdet/configs/point_rend/point_rend_r50_caffe_fpn_mstrain_3x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './point_rend_r50_caffe_fpn_mstrain_1x_coco.py' 2 | # learning policy 3 | lr_config = dict(step=[28, 34]) 4 | runner = dict(type='EpochBasedRunner', max_epochs=36) 5 | -------------------------------------------------------------------------------- /software/mmdet/configs/pvt/retinanet_pvt-m_fpn_1x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = 'retinanet_pvt-t_fpn_1x_coco.py' 2 | model = dict( 3 | backbone=dict( 4 | num_layers=[3, 4, 18, 3], 5 | init_cfg=dict(checkpoint='https://github.com/whai362/PVT/' 6 | 'releases/download/v2/pvt_medium.pth'))) 7 | -------------------------------------------------------------------------------- /software/mmdet/configs/pvt/retinanet_pvt-s_fpn_1x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = 'retinanet_pvt-t_fpn_1x_coco.py' 2 | model = dict( 3 | backbone=dict( 4 | num_layers=[3, 4, 6, 3], 5 | init_cfg=dict(checkpoint='https://github.com/whai362/PVT/' 6 | 'releases/download/v2/pvt_small.pth'))) 7 | -------------------------------------------------------------------------------- /software/mmdet/configs/queryinst/queryinst_r101_fpn_mstrain_480-800_3x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './queryinst_r50_fpn_mstrain_480-800_3x_coco.py' 2 | 3 | model = dict( 4 | backbone=dict( 5 | depth=101, 6 | init_cfg=dict(type='Pretrained', 7 | checkpoint='torchvision://resnet101'))) 8 | -------------------------------------------------------------------------------- /software/mmdet/configs/regnet/faster_rcnn_regnetx-3.2GF_fpn_2x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './faster_rcnn_regnetx-3.2GF_fpn_1x_coco.py' 2 | lr_config = dict(step=[16, 22]) 3 | runner = dict(type='EpochBasedRunner', max_epochs=24) 4 | -------------------------------------------------------------------------------- /software/mmdet/configs/reppoints/bbox_r50_grid_center_fpn_gn-neck+head_1x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './reppoints_moment_r50_fpn_gn-neck+head_1x_coco.py' 2 | model = dict(bbox_head=dict(transform_method='minmax', use_grid_points=True)) 3 | -------------------------------------------------------------------------------- /software/mmdet/configs/reppoints/reppoints_minmax_r50_fpn_gn-neck+head_1x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './reppoints_moment_r50_fpn_gn-neck+head_1x_coco.py' 2 | model = dict(bbox_head=dict(transform_method='minmax')) 3 | -------------------------------------------------------------------------------- /software/mmdet/configs/reppoints/reppoints_moment_r101_fpn_gn-neck+head_2x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './reppoints_moment_r50_fpn_gn-neck+head_2x_coco.py' 2 | model = dict( 3 | backbone=dict( 4 | depth=101, 5 | init_cfg=dict(type='Pretrained', 6 | checkpoint='torchvision://resnet101'))) 7 | -------------------------------------------------------------------------------- /software/mmdet/configs/reppoints/reppoints_moment_r50_fpn_gn-neck+head_1x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './reppoints_moment_r50_fpn_1x_coco.py' 2 | norm_cfg = dict(type='GN', num_groups=32, requires_grad=True) 3 | model = dict(neck=dict(norm_cfg=norm_cfg), bbox_head=dict(norm_cfg=norm_cfg)) 4 | optimizer = dict(lr=0.01) 5 | -------------------------------------------------------------------------------- /software/mmdet/configs/reppoints/reppoints_moment_r50_fpn_gn-neck+head_2x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './reppoints_moment_r50_fpn_gn-neck+head_1x_coco.py' 2 | lr_config = dict(step=[16, 22]) 3 | runner = dict(type='EpochBasedRunner', max_epochs=24) 4 | -------------------------------------------------------------------------------- /software/mmdet/configs/reppoints/reppoints_partial_minmax_r50_fpn_gn-neck+head_1x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './reppoints_moment_r50_fpn_gn-neck+head_1x_coco.py' 2 | model = dict(bbox_head=dict(transform_method='partial_minmax')) 3 | -------------------------------------------------------------------------------- /software/mmdet/configs/retinanet/retinanet_r101_caffe_fpn_1x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './retinanet_r50_caffe_fpn_1x_coco.py' 2 | model = dict( 3 | backbone=dict( 4 | depth=101, 5 | init_cfg=dict( 6 | type='Pretrained', 7 | checkpoint='open-mmlab://detectron2/resnet101_caffe'))) 8 | -------------------------------------------------------------------------------- /software/mmdet/configs/retinanet/retinanet_r101_fpn_1x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './retinanet_r50_fpn_1x_coco.py' 2 | model = dict( 3 | backbone=dict( 4 | depth=101, 5 | init_cfg=dict(type='Pretrained', 6 | checkpoint='torchvision://resnet101'))) 7 | -------------------------------------------------------------------------------- /software/mmdet/configs/retinanet/retinanet_r101_fpn_2x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './retinanet_r50_fpn_2x_coco.py' 2 | model = dict( 3 | backbone=dict( 4 | depth=101, 5 | init_cfg=dict(type='Pretrained', 6 | checkpoint='torchvision://resnet101'))) 7 | -------------------------------------------------------------------------------- /software/mmdet/configs/retinanet/retinanet_r50_caffe_fpn_mstrain_2x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './retinanet_r50_caffe_fpn_mstrain_1x_coco.py' 2 | # learning policy 3 | lr_config = dict(step=[16, 23]) 4 | runner = dict(type='EpochBasedRunner', max_epochs=24) 5 | -------------------------------------------------------------------------------- /software/mmdet/configs/retinanet/retinanet_r50_caffe_fpn_mstrain_3x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './retinanet_r50_caffe_fpn_mstrain_1x_coco.py' 2 | # learning policy 3 | lr_config = dict(step=[28, 34]) 4 | runner = dict(type='EpochBasedRunner', max_epochs=36) 5 | -------------------------------------------------------------------------------- /software/mmdet/configs/retinanet/retinanet_r50_fpn_2x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './retinanet_r50_fpn_1x_coco.py' 2 | # learning policy 3 | lr_config = dict(step=[16, 22]) 4 | runner = dict(type='EpochBasedRunner', max_epochs=24) 5 | -------------------------------------------------------------------------------- /software/mmdet/configs/retinanet/retinanet_r50_fpn_fp16_1x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './retinanet_r50_fpn_1x_coco.py' 2 | # fp16 settings 3 | fp16 = dict(loss_scale=512.) 4 | -------------------------------------------------------------------------------- /software/mmdet/configs/retinanet/retinanet_r50_fpn_mstrain_640-800_3x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/retinanet_r50_fpn.py', '../common/mstrain_3x_coco.py' 3 | ] 4 | # optimizer 5 | optimizer = dict(type='SGD', lr=0.01, momentum=0.9, weight_decay=0.0001) 6 | -------------------------------------------------------------------------------- /software/mmdet/configs/rpn/rpn_r101_caffe_fpn_1x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './rpn_r50_caffe_fpn_1x_coco.py' 2 | model = dict( 3 | backbone=dict( 4 | depth=101, 5 | init_cfg=dict( 6 | type='Pretrained', 7 | checkpoint='open-mmlab://detectron2/resnet101_caffe'))) 8 | -------------------------------------------------------------------------------- /software/mmdet/configs/rpn/rpn_r101_fpn_1x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './rpn_r50_fpn_1x_coco.py' 2 | model = dict( 3 | backbone=dict( 4 | depth=101, 5 | init_cfg=dict(type='Pretrained', 6 | checkpoint='torchvision://resnet101'))) 7 | -------------------------------------------------------------------------------- /software/mmdet/configs/rpn/rpn_r101_fpn_2x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './rpn_r50_fpn_2x_coco.py' 2 | model = dict( 3 | backbone=dict( 4 | depth=101, 5 | init_cfg=dict(type='Pretrained', 6 | checkpoint='torchvision://resnet101'))) 7 | -------------------------------------------------------------------------------- /software/mmdet/configs/rpn/rpn_r50_fpn_2x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './rpn_r50_fpn_1x_coco.py' 2 | 3 | # learning policy 4 | lr_config = dict(step=[16, 22]) 5 | runner = dict(type='EpochBasedRunner', max_epochs=24) 6 | -------------------------------------------------------------------------------- /software/mmdet/configs/scnet/scnet_r101_fpn_20e_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './scnet_r50_fpn_20e_coco.py' 2 | model = dict( 3 | backbone=dict( 4 | depth=101, 5 | init_cfg=dict(type='Pretrained', 6 | checkpoint='torchvision://resnet101'))) 7 | -------------------------------------------------------------------------------- /software/mmdet/configs/scnet/scnet_r50_fpn_20e_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './scnet_r50_fpn_1x_coco.py' 2 | # learning policy 3 | lr_config = dict(step=[16, 19]) 4 | runner = dict(type='EpochBasedRunner', max_epochs=20) 5 | -------------------------------------------------------------------------------- /software/mmdet/configs/seesaw_loss/mask_rcnn_r50_fpn_random_seesaw_loss_normed_mask_mstrain_2x_lvis_v1.py: -------------------------------------------------------------------------------- 1 | _base_ = './mask_rcnn_r50_fpn_random_seesaw_loss_mstrain_2x_lvis_v1.py' 2 | model = dict( 3 | roi_head=dict( 4 | mask_head=dict( 5 | predictor_cfg=dict(type='NormedConv2d', tempearture=20)))) 6 | -------------------------------------------------------------------------------- /software/mmdet/configs/seesaw_loss/mask_rcnn_r50_fpn_sample1e-3_seesaw_loss_normed_mask_mstrain_2x_lvis_v1.py: -------------------------------------------------------------------------------- 1 | _base_ = './mask_rcnn_r50_fpn_sample1e-3_seesaw_loss_mstrain_2x_lvis_v1.py' 2 | model = dict( 3 | roi_head=dict( 4 | mask_head=dict( 5 | predictor_cfg=dict(type='NormedConv2d', tempearture=20)))) 6 | -------------------------------------------------------------------------------- /software/mmdet/configs/solov2/solov2_light_r18_fpn_3x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = 'solov2_light_r50_fpn_3x_coco.py' 2 | 3 | # model settings 4 | model = dict( 5 | backbone=dict( 6 | depth=18, init_cfg=dict(checkpoint='torchvision://resnet18')), 7 | neck=dict(in_channels=[64, 128, 256, 512])) 8 | -------------------------------------------------------------------------------- /software/mmdet/configs/solov2/solov2_light_r34_fpn_3x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = 'solov2_light_r50_fpn_3x_coco.py' 2 | 3 | # model settings 4 | model = dict( 5 | backbone=dict( 6 | depth=34, init_cfg=dict(checkpoint='torchvision://resnet34')), 7 | neck=dict(in_channels=[64, 128, 256, 512])) 8 | -------------------------------------------------------------------------------- /software/mmdet/configs/solov2/solov2_r101_fpn_3x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = 'solov2_r50_fpn_3x_coco.py' 2 | 3 | # model settings 4 | model = dict( 5 | backbone=dict( 6 | depth=101, init_cfg=dict(checkpoint='torchvision://resnet101'))) 7 | -------------------------------------------------------------------------------- /software/mmdet/configs/sparse_rcnn/sparse_rcnn_r101_fpn_mstrain_480-800_3x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './sparse_rcnn_r50_fpn_mstrain_480-800_3x_coco.py' 2 | 3 | model = dict( 4 | backbone=dict( 5 | depth=101, 6 | init_cfg=dict(type='Pretrained', 7 | checkpoint='torchvision://resnet101'))) 8 | -------------------------------------------------------------------------------- /software/mmdet/configs/strong_baselines/mask_rcnn_r50_caffe_fpn_syncbn-all_rpn-2conv_lsj_100e_fp16_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = 'mask_rcnn_r50_caffe_fpn_syncbn-all_rpn-2conv_lsj_100e_coco.py' 2 | fp16 = dict(loss_scale=512.) 3 | -------------------------------------------------------------------------------- /software/mmdet/configs/strong_baselines/mask_rcnn_r50_fpn_syncbn-all_rpn-2conv_lsj_100e_fp16_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = 'mask_rcnn_r50_fpn_syncbn-all_rpn-2conv_lsj_100e_coco.py' 2 | # use FP16 3 | fp16 = dict(loss_scale=512.) 4 | -------------------------------------------------------------------------------- /software/mmdet/configs/strong_baselines/mask_rcnn_r50_fpn_syncbn-all_rpn-2conv_lsj_50e_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = 'mask_rcnn_r50_fpn_syncbn-all_rpn-2conv_lsj_100e_coco.py' 2 | 3 | # Use RepeatDataset to speed up training 4 | # change repeat time from 4 (for 100 epochs) to 2 (for 50 epochs) 5 | data = dict(train=dict(times=2)) 6 | -------------------------------------------------------------------------------- /software/mmdet/configs/swin/mask_rcnn_swin-t-p4-w7_fpn_fp16_ms-crop-3x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './mask_rcnn_swin-t-p4-w7_fpn_ms-crop-3x_coco.py' 2 | # you need to set mode='dynamic' if you are using pytorch<=1.5.0 3 | fp16 = dict(loss_scale=dict(init_scale=512)) 4 | -------------------------------------------------------------------------------- /software/mmdet/configs/tood/tood_r101_fpn_mstrain_2x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './tood_r50_fpn_mstrain_2x_coco.py' 2 | 3 | model = dict( 4 | backbone=dict( 5 | depth=101, 6 | init_cfg=dict(type='Pretrained', 7 | checkpoint='torchvision://resnet101'))) 8 | -------------------------------------------------------------------------------- /software/mmdet/configs/tood/tood_r50_fpn_anchor_based_1x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './tood_r50_fpn_1x_coco.py' 2 | model = dict(bbox_head=dict(anchor_type='anchor_based')) 3 | -------------------------------------------------------------------------------- /software/mmdet/configs/tridentnet/tridentnet_r50_caffe_mstrain_3x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = 'tridentnet_r50_caffe_mstrain_1x_coco.py' 2 | 3 | lr_config = dict(step=[28, 34]) 4 | runner = dict(type='EpochBasedRunner', max_epochs=36) 5 | -------------------------------------------------------------------------------- /software/mmdet/configs/vfnet/vfnet_r101_fpn_1x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './vfnet_r50_fpn_1x_coco.py' 2 | model = dict( 3 | backbone=dict( 4 | depth=101, 5 | init_cfg=dict(type='Pretrained', 6 | checkpoint='torchvision://resnet101'))) 7 | -------------------------------------------------------------------------------- /software/mmdet/configs/vfnet/vfnet_r101_fpn_mstrain_2x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './vfnet_r50_fpn_mstrain_2x_coco.py' 2 | model = dict( 3 | backbone=dict( 4 | depth=101, 5 | init_cfg=dict(type='Pretrained', 6 | checkpoint='torchvision://resnet101'))) 7 | -------------------------------------------------------------------------------- /software/mmdet/configs/yolact/yolact_r101_1x8_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './yolact_r50_1x8_coco.py' 2 | 3 | model = dict( 4 | backbone=dict( 5 | depth=101, 6 | init_cfg=dict(type='Pretrained', 7 | checkpoint='torchvision://resnet101'))) 8 | -------------------------------------------------------------------------------- /software/mmdet/configs/yolo/yolov3_d53_fp16_mstrain-608_273e_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = './yolov3_d53_mstrain-608_273e_coco.py' 2 | # fp16 settings 3 | fp16 = dict(loss_scale='dynamic') 4 | -------------------------------------------------------------------------------- /software/mmdet/demo/demo.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenRobotLab/DORT/5c1f39259247813a5c340753dd1a274f253a39b7/software/mmdet/demo/demo.mp4 -------------------------------------------------------------------------------- /software/mmdet/docker/serve/config.properties: -------------------------------------------------------------------------------- 1 | inference_address=http://0.0.0.0:8080 2 | management_address=http://0.0.0.0:8081 3 | metrics_address=http://0.0.0.0:8082 4 | model_store=/home/model-server/model-store 5 | load_models=all 6 | -------------------------------------------------------------------------------- /software/mmdet/docker/serve/entrypoint.sh: -------------------------------------------------------------------------------- 1 | #!/bin/bash 2 | set -e 3 | 4 | if [[ "$1" = "serve" ]]; then 5 | shift 1 6 | torchserve --start --ts-config /home/model-server/config.properties 7 | else 8 | eval "$@" 9 | fi 10 | 11 | # prevent docker exit 12 | tail -f /dev/null 13 | -------------------------------------------------------------------------------- /software/mmdet/docs/en/_static/css/readthedocs.css: -------------------------------------------------------------------------------- 1 | .header-logo { 2 | background-image: url("../image/mmdet-logo.png"); 3 | background-size: 156px 40px; 4 | height: 40px; 5 | width: 156px; 6 | } 7 | -------------------------------------------------------------------------------- /software/mmdet/docs/en/switch_language.md: -------------------------------------------------------------------------------- 1 | ## English 2 | 3 | ## 简体中文 4 | -------------------------------------------------------------------------------- /software/mmdet/docs/zh_cn/_static/css/readthedocs.css: -------------------------------------------------------------------------------- 1 | .header-logo { 2 | background-image: url("../image/mmdet-logo.png"); 3 | background-size: 156px 40px; 4 | height: 40px; 5 | width: 156px; 6 | } 7 | -------------------------------------------------------------------------------- /software/mmdet/docs/zh_cn/switch_language.md: -------------------------------------------------------------------------------- 1 | ## English 2 | 3 | ## 简体中文 4 | -------------------------------------------------------------------------------- /software/mmdet/docs/zh_cn/tutorials/customize_runtime.md: -------------------------------------------------------------------------------- 1 | # 教程 5: 自定义训练配置 2 | -------------------------------------------------------------------------------- /software/mmdet/docs/zh_cn/tutorials/pytorch2onnx.md: -------------------------------------------------------------------------------- 1 | # 教程 8: Pytorch 到 ONNX 的模型转换(实验性支持) 2 | 3 | > ## [尝试使用新的 MMDeploy 來部署你的模型](https://mmdeploy.readthedocs.io/) 4 | -------------------------------------------------------------------------------- /software/mmdet/docs/zh_cn/useful_tools.md: -------------------------------------------------------------------------------- 1 | ## 日志分析 2 | -------------------------------------------------------------------------------- /software/mmdet/mmdet/.mim/configs: -------------------------------------------------------------------------------- 1 | ../../configs -------------------------------------------------------------------------------- /software/mmdet/mmdet/.mim/demo: -------------------------------------------------------------------------------- 1 | ../../demo -------------------------------------------------------------------------------- /software/mmdet/mmdet/.mim/model-index.yml: -------------------------------------------------------------------------------- 1 | ../../model-index.yml -------------------------------------------------------------------------------- /software/mmdet/mmdet/.mim/tools: -------------------------------------------------------------------------------- 1 | ../../tools -------------------------------------------------------------------------------- /software/mmdet/mmdet/core/bbox/iou_calculators/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) OpenMMLab. All rights reserved. 2 | from .builder import build_iou_calculator 3 | from .iou2d_calculator import BboxOverlaps2D, bbox_overlaps 4 | 5 | __all__ = ['build_iou_calculator', 'BboxOverlaps2D', 'bbox_overlaps'] 6 | -------------------------------------------------------------------------------- /software/mmdet/mmdet/core/data_structures/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) OpenMMLab. All rights reserved. 2 | from .general_data import GeneralData 3 | from .instance_data import InstanceData 4 | 5 | __all__ = ['GeneralData', 'InstanceData'] 6 | -------------------------------------------------------------------------------- /software/mmdet/mmdet/models/roi_heads/shared_heads/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) OpenMMLab. All rights reserved. 2 | from .res_layer import ResLayer 3 | 4 | __all__ = ['ResLayer'] 5 | -------------------------------------------------------------------------------- /software/mmdet/mmdet/models/seg_heads/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) OpenMMLab. All rights reserved. 2 | from .panoptic_fpn_head import PanopticFPNHead # noqa: F401,F403 3 | from .panoptic_fusion_heads import * # noqa: F401,F403 4 | -------------------------------------------------------------------------------- /software/mmdet/requirements.txt: -------------------------------------------------------------------------------- 1 | -r requirements/build.txt 2 | -r requirements/optional.txt 3 | -r requirements/runtime.txt 4 | -r requirements/tests.txt 5 | -------------------------------------------------------------------------------- /software/mmdet/requirements/albu.txt: -------------------------------------------------------------------------------- 1 | albumentations>=0.3.2 --no-binary qudida,albumentations 2 | -------------------------------------------------------------------------------- /software/mmdet/requirements/build.txt: -------------------------------------------------------------------------------- 1 | # These must be installed before building mmdetection 2 | cython 3 | numpy 4 | -------------------------------------------------------------------------------- /software/mmdet/requirements/docs.txt: -------------------------------------------------------------------------------- 1 | docutils==0.16.0 2 | myst-parser 3 | -e git+https://github.com/open-mmlab/pytorch_sphinx_theme.git#egg=pytorch_sphinx_theme 4 | sphinx==4.0.2 5 | sphinx-copybutton 6 | sphinx_markdown_tables 7 | sphinx_rtd_theme==0.5.2 8 | -------------------------------------------------------------------------------- /software/mmdet/requirements/mminstall.txt: -------------------------------------------------------------------------------- 1 | mmcv-full>=1.3.17 2 | -------------------------------------------------------------------------------- /software/mmdet/requirements/optional.txt: -------------------------------------------------------------------------------- 1 | cityscapesscripts 2 | imagecorruptions 3 | scipy 4 | sklearn 5 | timm 6 | -------------------------------------------------------------------------------- /software/mmdet/requirements/readthedocs.txt: -------------------------------------------------------------------------------- 1 | mmcv 2 | torch 3 | torchvision 4 | -------------------------------------------------------------------------------- /software/mmdet/requirements/runtime.txt: -------------------------------------------------------------------------------- 1 | matplotlib 2 | numpy 3 | pycocotools 4 | six 5 | terminaltables 6 | -------------------------------------------------------------------------------- /software/mmdet/tests/test_data/test_pipelines/test_transform/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) OpenMMLab. All rights reserved. 2 | from .utils import check_result_same, construct_toy_data, create_random_bboxes 3 | 4 | __all__ = ['create_random_bboxes', 'construct_toy_data', 'check_result_same'] 5 | -------------------------------------------------------------------------------- /software/mmdet/tests/test_models/test_backbones/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) OpenMMLab. All rights reserved. 2 | from .utils import check_norm_state, is_block, is_norm 3 | 4 | __all__ = ['is_block', 'is_norm', 'check_norm_state'] 5 | -------------------------------------------------------------------------------- /software/mmdet/tests/test_models/test_roi_heads/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) OpenMMLab. All rights reserved. 2 | from .utils import _dummy_bbox_sampling 3 | 4 | __all__ = ['_dummy_bbox_sampling'] 5 | -------------------------------------------------------------------------------- /software/mmdet/tests/test_onnx/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) OpenMMLab. All rights reserved. 2 | from .utils import ort_validate 3 | 4 | __all__ = ['ort_validate'] 5 | -------------------------------------------------------------------------------- /software/mmdet3d/.dev_scripts/linter.sh: -------------------------------------------------------------------------------- 1 | yapf -r -i mmdet3d/ configs/ tests/ tools/ 2 | isort mmdet3d/ configs/ tests/ tools/ 3 | flake8 . 4 | -------------------------------------------------------------------------------- /software/mmdet3d/.github/CONTRIBUTING.md: -------------------------------------------------------------------------------- 1 | We appreciate all contributions to improve MMDetection3D. Please refer to [CONTRIBUTING.md](https://github.com/open-mmlab/mmcv/blob/master/CONTRIBUTING.md) in MMCV for more details about the contributing guideline. 2 | -------------------------------------------------------------------------------- /software/mmdet3d/.github/ISSUE_TEMPLATE/config.yml: -------------------------------------------------------------------------------- 1 | blank_issues_enabled: false 2 | -------------------------------------------------------------------------------- /software/mmdet3d/.github/ISSUE_TEMPLATE/general_questions.md: -------------------------------------------------------------------------------- 1 | --- 2 | name: General questions 3 | about: Ask general questions to get help 4 | title: '' 5 | labels: '' 6 | assignees: '' 7 | --- 8 | -------------------------------------------------------------------------------- /software/mmdet3d/.readthedocs.yml: -------------------------------------------------------------------------------- 1 | version: 2 2 | 3 | formats: all 4 | 5 | python: 6 | version: 3.7 7 | install: 8 | - requirements: requirements/docs.txt 9 | - requirements: requirements/runtime.txt 10 | - requirements: requirements/readthedocs.txt 11 | -------------------------------------------------------------------------------- /software/mmdet3d/MANIFEST.in: -------------------------------------------------------------------------------- 1 | include mmdet3d/.mim/model-index.yml 2 | include requirements/*.txt 3 | recursive-include mmdet3d/.mim/ops *.cpp *.cu *.h *.cc 4 | recursive-include mmdet3d/.mim/configs *.py *.yml 5 | recursive-include mmdet3d/.mim/tools *.sh *.py 6 | -------------------------------------------------------------------------------- /software/mmdet3d/configs/_base_/models/paconv_cuda_ssg.py: -------------------------------------------------------------------------------- 1 | _base_ = './paconv_ssg.py' 2 | 3 | model = dict( 4 | backbone=dict( 5 | sa_cfg=dict( 6 | type='PAConvCUDASAModule', 7 | scorenet_cfg=dict(mlp_channels=[8, 16, 16])))) 8 | -------------------------------------------------------------------------------- /software/mmdet3d/configs/centerpoint/centerpoint_0075voxel_second_secfpn_circlenms_4x8_cyclic_20e_nus.py: -------------------------------------------------------------------------------- 1 | _base_ = ['./centerpoint_0075voxel_second_secfpn_4x8_cyclic_20e_nus.py'] 2 | 3 | model = dict(test_cfg=dict(pts=dict(nms_type='circle'))) 4 | -------------------------------------------------------------------------------- /software/mmdet3d/configs/centerpoint/centerpoint_01voxel_second_secfpn_circlenms_4x8_cyclic_20e_nus.py: -------------------------------------------------------------------------------- 1 | _base_ = ['./centerpoint_01voxel_second_secfpn_4x8_cyclic_20e_nus.py'] 2 | 3 | model = dict(test_cfg=dict(pts=dict(nms_type='circle'))) 4 | -------------------------------------------------------------------------------- /software/mmdet3d/configs/centerpoint/centerpoint_02pillar_second_secfpn_circlenms_4x8_cyclic_20e_nus.py: -------------------------------------------------------------------------------- 1 | _base_ = ['./centerpoint_02pillar_second_secfpn_4x8_cyclic_20e_nus.py'] 2 | 3 | model = dict(test_cfg=dict(pts=dict(nms_type='circle'))) 4 | -------------------------------------------------------------------------------- /software/mmdet3d/configs/nuimages/cascade_mask_rcnn_r101_fpn_1x_nuim.py: -------------------------------------------------------------------------------- 1 | _base_ = './cascade_mask_rcnn_r50_fpn_1x_nuim.py' 2 | model = dict(pretrained='torchvision://resnet101', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmdet3d/configs/nuimages/htc_r50_fpn_coco-20e_1x_nuim.py: -------------------------------------------------------------------------------- 1 | _base_ = './htc_r50_fpn_1x_nuim.py' 2 | 3 | load_from = 'http://download.openmmlab.com/mmdetection/v2.0/htc/htc_r50_fpn_20e_coco/htc_r50_fpn_20e_coco_20200319-fe28c577.pth' # noqa 4 | -------------------------------------------------------------------------------- /software/mmdet3d/configs/nuimages/htc_r50_fpn_coco-20e_20e_nuim.py: -------------------------------------------------------------------------------- 1 | _base_ = './htc_r50_fpn_coco-20e_1x_nuim.py' 2 | # learning policy 3 | lr_config = dict(step=[16, 19]) 4 | runner = dict(max_epochs=20) 5 | -------------------------------------------------------------------------------- /software/mmdet3d/configs/nuimages/mask_rcnn_r101_fpn_1x_nuim.py: -------------------------------------------------------------------------------- 1 | _base_ = './mask_rcnn_r50_fpn_1x_nuim.py' 2 | model = dict(pretrained='torchvision://resnet101', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmdet3d/configs/pgd/pgd_r101_caffe_fpn_gn-head_2x16_2x_nus-mono3d.py: -------------------------------------------------------------------------------- 1 | _base_ = './pgd_r101_caffe_fpn_gn-head_2x16_1x_nus-mono3d.py' 2 | # learning policy 3 | lr_config = dict(step=[16, 22]) 4 | total_epochs = 24 5 | runner = dict(max_epochs=total_epochs) 6 | -------------------------------------------------------------------------------- /software/mmdet3d/configs/pointpillars/hv_pointpillars_fpn_sbn-all_2x8_2x_lyft-3d.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/hv_pointpillars_fpn_lyft.py', 3 | '../_base_/datasets/lyft-3d.py', '../_base_/schedules/schedule_2x.py', 4 | '../_base_/default_runtime.py' 5 | ] 6 | -------------------------------------------------------------------------------- /software/mmdet3d/configs/pointpillars/hv_pointpillars_fpn_sbn-all_4x8_2x_nus-3d.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/hv_pointpillars_fpn_nus.py', 3 | '../_base_/datasets/nus-3d.py', '../_base_/schedules/schedule_2x.py', 4 | '../_base_/default_runtime.py' 5 | ] 6 | -------------------------------------------------------------------------------- /software/mmdet3d/configs/pointpillars/hv_pointpillars_fpn_sbn-all_fp16_2x8_2x_nus-3d.py: -------------------------------------------------------------------------------- 1 | _base_ = './hv_pointpillars_fpn_sbn-all_4x8_2x_nus-3d.py' 2 | data = dict(samples_per_gpu=2, workers_per_gpu=2) 3 | # fp16 settings, the loss scale is specifically tuned to avoid Nan 4 | fp16 = dict(loss_scale=32.) 5 | -------------------------------------------------------------------------------- /software/mmdet3d/configs/pointpillars/hv_pointpillars_fpn_sbn-all_range100_2x8_2x_lyft-3d.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/hv_pointpillars_fpn_range100_lyft.py', 3 | '../_base_/datasets/range100_lyft-3d.py', 4 | '../_base_/schedules/schedule_2x.py', '../_base_/default_runtime.py' 5 | ] 6 | -------------------------------------------------------------------------------- /software/mmdet3d/configs/pointpillars/hv_pointpillars_secfpn_sbn-all_fp16_2x8_2x_nus-3d.py: -------------------------------------------------------------------------------- 1 | _base_ = './hv_pointpillars_secfpn_sbn-all_4x8_2x_nus-3d.py' 2 | data = dict(samples_per_gpu=2, workers_per_gpu=2) 3 | # fp16 settings, the loss scale is specifically tuned to avoid Nan 4 | fp16 = dict(loss_scale=32.) 5 | -------------------------------------------------------------------------------- /software/mmdet3d/configs/pointpillars/hv_pointpillars_secfpn_sbn_2x16_2x_waymoD5-3d-3class.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/hv_pointpillars_secfpn_waymo.py', 3 | '../_base_/datasets/waymoD5-3d-3class.py', 4 | '../_base_/schedules/schedule_2x.py', 5 | '../_base_/default_runtime.py', 6 | ] 7 | -------------------------------------------------------------------------------- /software/mmdet3d/configs/regnet/hv_pointpillars_regnet-400mf_fpn_sbn-all_fp16_2x8_2x_nus-3d.py: -------------------------------------------------------------------------------- 1 | _base_ = './hv_pointpillars_regnet-400mf_fpn_sbn-all_4x8_2x_nus-3d.py' 2 | data = dict(samples_per_gpu=2, workers_per_gpu=2) 3 | # fp16 settings, the loss scale is specifically tuned to avoid Nan 4 | fp16 = dict(loss_scale=32.) 5 | -------------------------------------------------------------------------------- /software/mmdet3d/configs/second/hv_second_secfpn_6x8_80e_kitti-3d-3class.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/hv_second_secfpn_kitti.py', 3 | '../_base_/datasets/kitti-3d-3class.py', 4 | '../_base_/schedules/cyclic_40e.py', '../_base_/default_runtime.py' 5 | ] 6 | -------------------------------------------------------------------------------- /software/mmdet3d/configs/second/hv_second_secfpn_fp16_6x8_80e_kitti-3d-3class.py: -------------------------------------------------------------------------------- 1 | _base_ = './hv_second_secfpn_6x8_80e_kitti-3d-3class.py' 2 | # fp16 settings 3 | fp16 = dict(loss_scale=512.) 4 | -------------------------------------------------------------------------------- /software/mmdet3d/configs/second/hv_second_secfpn_fp16_6x8_80e_kitti-3d-car.py: -------------------------------------------------------------------------------- 1 | _base_ = './hv_second_secfpn_6x8_80e_kitti-3d-car.py' 2 | # fp16 settings 3 | fp16 = dict(loss_scale=512.) 4 | -------------------------------------------------------------------------------- /software/mmdet3d/docker/serve/config.properties: -------------------------------------------------------------------------------- 1 | inference_address=http://0.0.0.0:8080 2 | management_address=http://0.0.0.0:8081 3 | metrics_address=http://0.0.0.0:8082 4 | model_store=/home/model-server/model-store 5 | load_models=all 6 | -------------------------------------------------------------------------------- /software/mmdet3d/docker/serve/entrypoint.sh: -------------------------------------------------------------------------------- 1 | #!/bin/bash 2 | set -e 3 | 4 | if [[ "$1" = "serve" ]]; then 5 | shift 1 6 | torchserve --start --ts-config /home/model-server/config.properties 7 | else 8 | eval "$@" 9 | fi 10 | 11 | # prevent docker exit 12 | tail -f /dev/null 13 | -------------------------------------------------------------------------------- /software/mmdet3d/docs/en/_static/css/readthedocs.css: -------------------------------------------------------------------------------- 1 | .header-logo { 2 | background-image: url("../image/mmdet3d-logo.png"); 3 | background-size: 182.5px 40px; 4 | height: 40px; 5 | width: 182.5px; 6 | } 7 | -------------------------------------------------------------------------------- /software/mmdet3d/docs/en/datasets/index.rst: -------------------------------------------------------------------------------- 1 | .. toctree:: 2 | :maxdepth: 2 3 | 4 | kitti_det.md 5 | nuscenes_det.md 6 | lyft_det.md 7 | waymo_det.md 8 | sunrgbd_det.md 9 | scannet_det.md 10 | scannet_sem_seg.md 11 | s3dis_sem_seg.md 12 | -------------------------------------------------------------------------------- /software/mmdet3d/docs/en/supported_tasks/index.rst: -------------------------------------------------------------------------------- 1 | .. toctree:: 2 | :maxdepth: 2 3 | 4 | lidar_det3d.md 5 | vision_det3d.md 6 | lidar_sem_seg3d.md 7 | -------------------------------------------------------------------------------- /software/mmdet3d/docs/en/switch_language.md: -------------------------------------------------------------------------------- 1 | ## English 2 | 3 | ## 简体中文 4 | -------------------------------------------------------------------------------- /software/mmdet3d/docs/zh_cn/_static/css/readthedocs.css: -------------------------------------------------------------------------------- 1 | .header-logo { 2 | background-image: url("../image/mmdet3d-logo.png"); 3 | background-size: 182.5px 40px; 4 | height: 40px; 5 | width: 182.5px; 6 | } 7 | -------------------------------------------------------------------------------- /software/mmdet3d/docs/zh_cn/changelog.md: -------------------------------------------------------------------------------- 1 | # 变更日志 2 | -------------------------------------------------------------------------------- /software/mmdet3d/docs/zh_cn/compatibility.md: -------------------------------------------------------------------------------- 1 | ## 0.16.0 2 | -------------------------------------------------------------------------------- /software/mmdet3d/docs/zh_cn/datasets/index.rst: -------------------------------------------------------------------------------- 1 | .. toctree:: 2 | :maxdepth: 2 3 | 4 | kitti_det.md 5 | nuscenes_det.md 6 | lyft_det.md 7 | waymo_det.md 8 | sunrgbd_det.md 9 | scannet_det.md 10 | scannet_sem_seg.md 11 | s3dis_sem_seg.md 12 | -------------------------------------------------------------------------------- /software/mmdet3d/docs/zh_cn/supported_tasks/index.rst: -------------------------------------------------------------------------------- 1 | .. toctree:: 2 | :maxdepth: 2 3 | 4 | lidar_det3d.md 5 | vision_det3d.md 6 | lidar_sem_seg3d.md 7 | -------------------------------------------------------------------------------- /software/mmdet3d/docs/zh_cn/switch_language.md: -------------------------------------------------------------------------------- 1 | ## English 2 | 3 | ## 简体中文 4 | -------------------------------------------------------------------------------- /software/mmdet3d/docs/zh_cn/tutorials/index.rst: -------------------------------------------------------------------------------- 1 | .. toctree:: 2 | :maxdepth: 2 3 | 4 | config.md 5 | customize_dataset.md 6 | data_pipeline.md 7 | customize_models.md 8 | customize_runtime.md 9 | coord_sys_tutorial.md 10 | backends_support.md 11 | -------------------------------------------------------------------------------- /software/mmdet3d/mmdet3d/core/bbox/assigners/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) OpenMMLab. All rights reserved. 2 | from mmdet.core.bbox import AssignResult, BaseAssigner, MaxIoUAssigner 3 | 4 | __all__ = ['BaseAssigner', 'MaxIoUAssigner', 'AssignResult'] 5 | -------------------------------------------------------------------------------- /software/mmdet3d/mmdet3d/core/evaluation/kitti_utils/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) OpenMMLab. All rights reserved. 2 | from .eval import kitti_eval, kitti_eval_coco_style 3 | 4 | __all__ = ['kitti_eval', 'kitti_eval_coco_style'] 5 | -------------------------------------------------------------------------------- /software/mmdet3d/mmdet3d/core/evaluation/scannet_utils/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) OpenMMLab. All rights reserved. 2 | from .evaluate_semantic_instance import evaluate_matches, scannet_eval 3 | 4 | __all__ = ['scannet_eval', 'evaluate_matches'] 5 | -------------------------------------------------------------------------------- /software/mmdet3d/mmdet3d/core/evaluation/waymo_utils/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) OpenMMLab. All rights reserved. 2 | from .prediction_kitti_to_waymo import KITTI2Waymo 3 | 4 | __all__ = ['KITTI2Waymo'] 5 | -------------------------------------------------------------------------------- /software/mmdet3d/mmdet3d/core/evaluation/waymo_utils/compute_detection_let_metrics_main: -------------------------------------------------------------------------------- 1 | /mnt/lustre/lianqing/data/waymo/eval_utils/compute_detection_let_metrics_main -------------------------------------------------------------------------------- /software/mmdet3d/mmdet3d/core/visualizer/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) OpenMMLab. All rights reserved. 2 | from .show_result import (show_multi_modality_result, show_result, 3 | show_seg_result) 4 | 5 | __all__ = ['show_result', 'show_seg_result', 'show_multi_modality_result'] 6 | -------------------------------------------------------------------------------- /software/mmdet3d/mmdet3d/core/voxel/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) OpenMMLab. All rights reserved. 2 | from .builder import build_voxel_generator 3 | from .voxel_generator import VoxelGenerator 4 | 5 | __all__ = ['build_voxel_generator', 'VoxelGenerator'] 6 | -------------------------------------------------------------------------------- /software/mmdet3d/mmdet3d/models/decode_heads/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) OpenMMLab. All rights reserved. 2 | from .dgcnn_head import DGCNNHead 3 | from .paconv_head import PAConvHead 4 | from .pointnet2_head import PointNet2Head 5 | 6 | __all__ = ['PointNet2Head', 'DGCNNHead', 'PAConvHead'] 7 | -------------------------------------------------------------------------------- /software/mmdet3d/mmdet3d/models/model_utils/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) OpenMMLab. All rights reserved. 2 | from .edge_fusion_module import EdgeFusionModule 3 | from .transformer import GroupFree3DMHA 4 | from .vote_module import VoteModule 5 | 6 | __all__ = ['VoteModule', 'GroupFree3DMHA', 'EdgeFusionModule'] 7 | -------------------------------------------------------------------------------- /software/mmdet3d/mmdet3d/models/roi_heads/mask_heads/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) OpenMMLab. All rights reserved. 2 | from .pointwise_semantic_head import PointwiseSemanticHead 3 | from .primitive_head import PrimitiveHead 4 | 5 | __all__ = ['PointwiseSemanticHead', 'PrimitiveHead'] 6 | -------------------------------------------------------------------------------- /software/mmdet3d/mmdet3d/models/segmentors/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) OpenMMLab. All rights reserved. 2 | from .base import Base3DSegmentor 3 | from .encoder_decoder import EncoderDecoder3D 4 | 5 | __all__ = ['Base3DSegmentor', 'EncoderDecoder3D'] 6 | -------------------------------------------------------------------------------- /software/mmdet3d/mmdet3d/ops/dgcnn_modules/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) OpenMMLab. All rights reserved. 2 | from .dgcnn_fa_module import DGCNNFAModule 3 | from .dgcnn_fp_module import DGCNNFPModule 4 | from .dgcnn_gf_module import DGCNNGFModule 5 | 6 | __all__ = ['DGCNNFAModule', 'DGCNNFPModule', 'DGCNNGFModule'] 7 | -------------------------------------------------------------------------------- /software/mmdet3d/mmdet3d/ops/paconv/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) OpenMMLab. All rights reserved. 2 | from .paconv import PAConv, PAConvCUDA 3 | 4 | __all__ = ['PAConv', 'PAConvCUDA'] 5 | -------------------------------------------------------------------------------- /software/mmdet3d/mmdet3d/ops/spconv/overwrite_spconv/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) OpenMMLab. All rights reserved. 2 | from .write_spconv2 import register_spconv2 3 | 4 | __all__ = ['register_spconv2'] 5 | -------------------------------------------------------------------------------- /software/mmdet3d/requirements.txt: -------------------------------------------------------------------------------- 1 | -r requirements/build.txt 2 | -r requirements/optional.txt 3 | -r requirements/runtime.txt 4 | -r requirements/tests.txt 5 | -------------------------------------------------------------------------------- /software/mmdet3d/requirements/build.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenRobotLab/DORT/5c1f39259247813a5c340753dd1a274f253a39b7/software/mmdet3d/requirements/build.txt -------------------------------------------------------------------------------- /software/mmdet3d/requirements/docs.txt: -------------------------------------------------------------------------------- 1 | docutils==0.16.0 2 | m2r 3 | mistune==0.8.4 4 | myst-parser 5 | -e git+https://github.com/open-mmlab/pytorch_sphinx_theme.git#egg=pytorch_sphinx_theme 6 | sphinx==4.0.2 7 | sphinx-copybutton 8 | sphinx_markdown_tables 9 | -------------------------------------------------------------------------------- /software/mmdet3d/requirements/mminstall.txt: -------------------------------------------------------------------------------- 1 | mmcv-full>=1.4.8,<=1.6.0 2 | mmdet>=2.24.0,<=3.0.0 3 | mmsegmentation>=0.20.0,<=1.0.0 4 | -------------------------------------------------------------------------------- /software/mmdet3d/requirements/optional.txt: -------------------------------------------------------------------------------- 1 | open3d 2 | spconv 3 | waymo-open-dataset-tf-2-1-0==1.2.0 4 | -------------------------------------------------------------------------------- /software/mmdet3d/requirements/readthedocs.txt: -------------------------------------------------------------------------------- 1 | mmcv>=1.4.8 2 | mmdet>=2.24.0 3 | mmsegmentation>=0.20.1 4 | torch 5 | torchvision 6 | -------------------------------------------------------------------------------- /software/mmdet3d/requirements/runtime.txt: -------------------------------------------------------------------------------- 1 | lyft_dataset_sdk 2 | networkx>=2.2,<2.3 3 | numba==0.53.0 4 | numpy 5 | nuscenes-devkit 6 | plyfile 7 | scikit-image 8 | # by default we also use tensorboard to log results 9 | tensorboard 10 | trimesh>=2.35.39,<2.35.40 11 | -------------------------------------------------------------------------------- /software/mmdet3d/requirements/tests.txt: -------------------------------------------------------------------------------- 1 | asynctest 2 | codecov 3 | flake8 4 | interrogate 5 | isort 6 | # Note: used for kwarray.group_items, this may be ported to mmcv in the future. 7 | kwarray 8 | pytest 9 | pytest-cov 10 | pytest-runner 11 | ubelt 12 | xdoctest >= 0.10.0 13 | yapf 14 | -------------------------------------------------------------------------------- /software/mmdet3d/resources/mmdet3d_outdoor_demo.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenRobotLab/DORT/5c1f39259247813a5c340753dd1a274f253a39b7/software/mmdet3d/resources/mmdet3d_outdoor_demo.gif -------------------------------------------------------------------------------- /software/mmdet3d/resources/nuimages_demo.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenRobotLab/DORT/5c1f39259247813a5c340753dd1a274f253a39b7/software/mmdet3d/resources/nuimages_demo.gif -------------------------------------------------------------------------------- /software/mmdet3d/resources/open3d_visual.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenRobotLab/DORT/5c1f39259247813a5c340753dd1a274f253a39b7/software/mmdet3d/resources/open3d_visual.gif -------------------------------------------------------------------------------- /software/mmdet3d/tests/test_samples/parta2_roihead_inputs.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenRobotLab/DORT/5c1f39259247813a5c340753dd1a274f253a39b7/software/mmdet3d/tests/test_samples/parta2_roihead_inputs.npz -------------------------------------------------------------------------------- /software/mmdet3d/tools/data_converter/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) OpenMMLab. All rights reserved. 2 | -------------------------------------------------------------------------------- /software/mmseg/.readthedocs.yml: -------------------------------------------------------------------------------- 1 | version: 2 2 | 3 | formats: all 4 | 5 | python: 6 | version: 3.7 7 | install: 8 | - requirements: requirements/docs.txt 9 | - requirements: requirements/readthedocs.txt 10 | -------------------------------------------------------------------------------- /software/mmseg/MANIFEST.in: -------------------------------------------------------------------------------- 1 | include requirements/*.txt 2 | include mmseg/.mim/model-index.yml 3 | recursive-include mmseg/.mim/configs *.py *.yml 4 | recursive-include mmseg/.mim/tools *.py *.sh 5 | -------------------------------------------------------------------------------- /software/mmseg/configs/ann/ann_r101-d8_512x1024_40k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './ann_r50-d8_512x1024_40k_cityscapes.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/ann/ann_r101-d8_512x1024_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './ann_r50-d8_512x1024_80k_cityscapes.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/ann/ann_r101-d8_512x512_160k_ade20k.py: -------------------------------------------------------------------------------- 1 | _base_ = './ann_r50-d8_512x512_160k_ade20k.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/ann/ann_r101-d8_512x512_20k_voc12aug.py: -------------------------------------------------------------------------------- 1 | _base_ = './ann_r50-d8_512x512_20k_voc12aug.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/ann/ann_r101-d8_512x512_40k_voc12aug.py: -------------------------------------------------------------------------------- 1 | _base_ = './ann_r50-d8_512x512_40k_voc12aug.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/ann/ann_r101-d8_512x512_80k_ade20k.py: -------------------------------------------------------------------------------- 1 | _base_ = './ann_r50-d8_512x512_80k_ade20k.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/ann/ann_r101-d8_769x769_40k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './ann_r50-d8_769x769_40k_cityscapes.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/ann/ann_r101-d8_769x769_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './ann_r50-d8_769x769_80k_cityscapes.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/ann/ann_r50-d8_512x1024_40k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/ann_r50-d8.py', '../_base_/datasets/cityscapes.py', 3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_40k.py' 4 | ] 5 | -------------------------------------------------------------------------------- /software/mmseg/configs/ann/ann_r50-d8_512x1024_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/ann_r50-d8.py', '../_base_/datasets/cityscapes.py', 3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_80k.py' 4 | ] 5 | -------------------------------------------------------------------------------- /software/mmseg/configs/apcnet/apcnet_r101-d8_512x1024_40k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './apcnet_r50-d8_512x1024_40k_cityscapes.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/apcnet/apcnet_r101-d8_512x1024_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './apcnet_r50-d8_512x1024_80k_cityscapes.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/apcnet/apcnet_r101-d8_512x512_160k_ade20k.py: -------------------------------------------------------------------------------- 1 | _base_ = './apcnet_r50-d8_512x512_160k_ade20k.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/apcnet/apcnet_r101-d8_512x512_80k_ade20k.py: -------------------------------------------------------------------------------- 1 | _base_ = './apcnet_r50-d8_512x512_80k_ade20k.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/apcnet/apcnet_r101-d8_769x769_40k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './apcnet_r50-d8_769x769_40k_cityscapes.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/apcnet/apcnet_r101-d8_769x769_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './apcnet_r50-d8_769x769_80k_cityscapes.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/apcnet/apcnet_r50-d8_512x1024_40k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/apcnet_r50-d8.py', '../_base_/datasets/cityscapes.py', 3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_40k.py' 4 | ] 5 | -------------------------------------------------------------------------------- /software/mmseg/configs/apcnet/apcnet_r50-d8_512x1024_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/apcnet_r50-d8.py', '../_base_/datasets/cityscapes.py', 3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_80k.py' 4 | ] 5 | -------------------------------------------------------------------------------- /software/mmseg/configs/bisenetv1/bisenetv1_r18-d32_in1k-pre_4x8_1024x1024_160k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './bisenetv1_r18-d32_in1k-pre_4x4_1024x1024_160k_cityscapes.py' 2 | data = dict( 3 | samples_per_gpu=8, 4 | workers_per_gpu=8, 5 | ) 6 | -------------------------------------------------------------------------------- /software/mmseg/configs/bisenetv2/bisenetv2_fcn_fp16_4x4_1024x1024_160k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './bisenetv2_fcn_4x4_1024x1024_160k_cityscapes.py' 2 | # fp16 settings 3 | optimizer_config = dict(type='Fp16OptimizerHook', loss_scale=512.) 4 | # fp16 placeholder 5 | fp16 = dict() 6 | -------------------------------------------------------------------------------- /software/mmseg/configs/ccnet/ccnet_r101-d8_512x1024_40k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './ccnet_r50-d8_512x1024_40k_cityscapes.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/ccnet/ccnet_r101-d8_512x1024_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './ccnet_r50-d8_512x1024_80k_cityscapes.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/ccnet/ccnet_r101-d8_512x512_160k_ade20k.py: -------------------------------------------------------------------------------- 1 | _base_ = './ccnet_r50-d8_512x512_160k_ade20k.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/ccnet/ccnet_r101-d8_512x512_20k_voc12aug.py: -------------------------------------------------------------------------------- 1 | _base_ = './ccnet_r50-d8_512x512_20k_voc12aug.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/ccnet/ccnet_r101-d8_512x512_40k_voc12aug.py: -------------------------------------------------------------------------------- 1 | _base_ = './ccnet_r50-d8_512x512_40k_voc12aug.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/ccnet/ccnet_r101-d8_512x512_80k_ade20k.py: -------------------------------------------------------------------------------- 1 | _base_ = './ccnet_r50-d8_512x512_80k_ade20k.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/ccnet/ccnet_r101-d8_769x769_40k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './ccnet_r50-d8_769x769_40k_cityscapes.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/ccnet/ccnet_r101-d8_769x769_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './ccnet_r50-d8_769x769_80k_cityscapes.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/ccnet/ccnet_r50-d8_512x1024_40k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/ccnet_r50-d8.py', '../_base_/datasets/cityscapes.py', 3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_40k.py' 4 | ] 5 | -------------------------------------------------------------------------------- /software/mmseg/configs/ccnet/ccnet_r50-d8_512x1024_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/ccnet_r50-d8.py', '../_base_/datasets/cityscapes.py', 3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_80k.py' 4 | ] 5 | -------------------------------------------------------------------------------- /software/mmseg/configs/danet/danet_r101-d8_512x1024_40k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './danet_r50-d8_512x1024_40k_cityscapes.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/danet/danet_r101-d8_512x1024_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './danet_r50-d8_512x1024_80k_cityscapes.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/danet/danet_r101-d8_512x512_160k_ade20k.py: -------------------------------------------------------------------------------- 1 | _base_ = './danet_r50-d8_512x512_160k_ade20k.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/danet/danet_r101-d8_512x512_20k_voc12aug.py: -------------------------------------------------------------------------------- 1 | _base_ = './danet_r50-d8_512x512_20k_voc12aug.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/danet/danet_r101-d8_512x512_40k_voc12aug.py: -------------------------------------------------------------------------------- 1 | _base_ = './danet_r50-d8_512x512_40k_voc12aug.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/danet/danet_r101-d8_512x512_80k_ade20k.py: -------------------------------------------------------------------------------- 1 | _base_ = './danet_r50-d8_512x512_80k_ade20k.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/danet/danet_r101-d8_769x769_40k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './danet_r50-d8_769x769_40k_cityscapes.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/danet/danet_r101-d8_769x769_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './danet_r50-d8_769x769_80k_cityscapes.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/danet/danet_r50-d8_512x1024_40k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/danet_r50-d8.py', '../_base_/datasets/cityscapes.py', 3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_40k.py' 4 | ] 5 | -------------------------------------------------------------------------------- /software/mmseg/configs/danet/danet_r50-d8_512x1024_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/danet_r50-d8.py', '../_base_/datasets/cityscapes.py', 3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_80k.py' 4 | ] 5 | -------------------------------------------------------------------------------- /software/mmseg/configs/deeplabv3/deeplabv3_r101-d8_480x480_40k_pascal_context.py: -------------------------------------------------------------------------------- 1 | _base_ = './deeplabv3_r50-d8_480x480_40k_pascal_context.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/deeplabv3/deeplabv3_r101-d8_480x480_40k_pascal_context_59.py: -------------------------------------------------------------------------------- 1 | _base_ = './deeplabv3_r50-d8_480x480_40k_pascal_context_59.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/deeplabv3/deeplabv3_r101-d8_480x480_80k_pascal_context.py: -------------------------------------------------------------------------------- 1 | _base_ = './deeplabv3_r50-d8_480x480_80k_pascal_context.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/deeplabv3/deeplabv3_r101-d8_480x480_80k_pascal_context_59.py: -------------------------------------------------------------------------------- 1 | _base_ = './deeplabv3_r50-d8_480x480_80k_pascal_context_59.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/deeplabv3/deeplabv3_r101-d8_512x1024_40k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './deeplabv3_r50-d8_512x1024_40k_cityscapes.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/deeplabv3/deeplabv3_r101-d8_512x1024_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './deeplabv3_r50-d8_512x1024_80k_cityscapes.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/deeplabv3/deeplabv3_r101-d8_512x512_160k_ade20k.py: -------------------------------------------------------------------------------- 1 | _base_ = './deeplabv3_r50-d8_512x512_160k_ade20k.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/deeplabv3/deeplabv3_r101-d8_512x512_20k_voc12aug.py: -------------------------------------------------------------------------------- 1 | _base_ = './deeplabv3_r50-d8_512x512_20k_voc12aug.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/deeplabv3/deeplabv3_r101-d8_512x512_40k_voc12aug.py: -------------------------------------------------------------------------------- 1 | _base_ = './deeplabv3_r50-d8_512x512_40k_voc12aug.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/deeplabv3/deeplabv3_r101-d8_512x512_4x4_160k_coco-stuff164k.py: -------------------------------------------------------------------------------- 1 | _base_ = './deeplabv3_r50-d8_512x512_4x4_160k_coco-stuff164k.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/deeplabv3/deeplabv3_r101-d8_512x512_4x4_20k_coco-stuff10k.py: -------------------------------------------------------------------------------- 1 | _base_ = './deeplabv3_r50-d8_512x512_4x4_20k_coco-stuff10k.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/deeplabv3/deeplabv3_r101-d8_512x512_4x4_320k_coco-stuff164k.py: -------------------------------------------------------------------------------- 1 | _base_ = './deeplabv3_r50-d8_512x512_4x4_320k_coco-stuff164k.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/deeplabv3/deeplabv3_r101-d8_512x512_4x4_40k_coco-stuff10k.py: -------------------------------------------------------------------------------- 1 | _base_ = './deeplabv3_r50-d8_512x512_4x4_40k_coco-stuff10k.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/deeplabv3/deeplabv3_r101-d8_512x512_4x4_80k_coco-stuff164k.py: -------------------------------------------------------------------------------- 1 | _base_ = './deeplabv3_r50-d8_512x512_4x4_80k_coco-stuff164k.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/deeplabv3/deeplabv3_r101-d8_512x512_80k_ade20k.py: -------------------------------------------------------------------------------- 1 | _base_ = './deeplabv3_r50-d8_512x512_80k_ade20k.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/deeplabv3/deeplabv3_r101-d8_769x769_40k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './deeplabv3_r50-d8_769x769_40k_cityscapes.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/deeplabv3/deeplabv3_r101-d8_769x769_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './deeplabv3_r50-d8_769x769_80k_cityscapes.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/deeplabv3/deeplabv3_r101-d8_fp16_512x1024_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './deeplabv3_r101-d8_512x1024_80k_cityscapes.py' 2 | # fp16 settings 3 | optimizer_config = dict(type='Fp16OptimizerHook', loss_scale=512.) 4 | # fp16 placeholder 5 | fp16 = dict() 6 | -------------------------------------------------------------------------------- /software/mmseg/configs/deeplabv3/deeplabv3_r101b-d8_512x1024_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './deeplabv3_r50-d8_512x1024_80k_cityscapes.py' 2 | model = dict( 3 | pretrained='torchvision://resnet101', 4 | backbone=dict(type='ResNet', depth=101)) 5 | -------------------------------------------------------------------------------- /software/mmseg/configs/deeplabv3/deeplabv3_r101b-d8_769x769_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './deeplabv3_r50-d8_769x769_80k_cityscapes.py' 2 | model = dict( 3 | pretrained='torchvision://resnet101', 4 | backbone=dict(type='ResNet', depth=101)) 5 | -------------------------------------------------------------------------------- /software/mmseg/configs/deeplabv3/deeplabv3_r50-d8_512x1024_40k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/deeplabv3_r50-d8.py', '../_base_/datasets/cityscapes.py', 3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_40k.py' 4 | ] 5 | -------------------------------------------------------------------------------- /software/mmseg/configs/deeplabv3/deeplabv3_r50-d8_512x1024_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/deeplabv3_r50-d8.py', '../_base_/datasets/cityscapes.py', 3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_80k.py' 4 | ] 5 | -------------------------------------------------------------------------------- /software/mmseg/configs/deeplabv3/deeplabv3_r50b-d8_512x1024_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './deeplabv3_r50-d8_512x1024_80k_cityscapes.py' 2 | model = dict(pretrained='torchvision://resnet50', backbone=dict(type='ResNet')) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/deeplabv3/deeplabv3_r50b-d8_769x769_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './deeplabv3_r50-d8_769x769_80k_cityscapes.py' 2 | model = dict(pretrained='torchvision://resnet50', backbone=dict(type='ResNet')) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/deeplabv3plus/deeplabv3plus_r101-d8_480x480_40k_pascal_context.py: -------------------------------------------------------------------------------- 1 | _base_ = './deeplabv3plus_r50-d8_480x480_40k_pascal_context.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/deeplabv3plus/deeplabv3plus_r101-d8_480x480_40k_pascal_context_59.py: -------------------------------------------------------------------------------- 1 | _base_ = './deeplabv3plus_r50-d8_480x480_40k_pascal_context_59.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/deeplabv3plus/deeplabv3plus_r101-d8_480x480_80k_pascal_context.py: -------------------------------------------------------------------------------- 1 | _base_ = './deeplabv3plus_r50-d8_480x480_80k_pascal_context.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/deeplabv3plus/deeplabv3plus_r101-d8_480x480_80k_pascal_context_59.py: -------------------------------------------------------------------------------- 1 | _base_ = './deeplabv3plus_r50-d8_480x480_80k_pascal_context_59.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/deeplabv3plus/deeplabv3plus_r101-d8_4x4_512x512_80k_vaihingen.py: -------------------------------------------------------------------------------- 1 | _base_ = './deeplabv3plus_r50-d8_4x4_512x512_80k_vaihingen.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/deeplabv3plus/deeplabv3plus_r101-d8_512x1024_40k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './deeplabv3plus_r50-d8_512x1024_40k_cityscapes.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/deeplabv3plus/deeplabv3plus_r101-d8_512x1024_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './deeplabv3plus_r50-d8_512x1024_80k_cityscapes.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/deeplabv3plus/deeplabv3plus_r101-d8_512x512_160k_ade20k.py: -------------------------------------------------------------------------------- 1 | _base_ = './deeplabv3plus_r50-d8_512x512_160k_ade20k.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/deeplabv3plus/deeplabv3plus_r101-d8_512x512_20k_voc12aug.py: -------------------------------------------------------------------------------- 1 | _base_ = './deeplabv3plus_r50-d8_512x512_20k_voc12aug.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/deeplabv3plus/deeplabv3plus_r101-d8_512x512_40k_voc12aug.py: -------------------------------------------------------------------------------- 1 | _base_ = './deeplabv3plus_r50-d8_512x512_40k_voc12aug.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/deeplabv3plus/deeplabv3plus_r101-d8_512x512_80k_ade20k.py: -------------------------------------------------------------------------------- 1 | _base_ = './deeplabv3plus_r50-d8_512x512_80k_ade20k.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/deeplabv3plus/deeplabv3plus_r101-d8_512x512_80k_loveda.py: -------------------------------------------------------------------------------- 1 | _base_ = './deeplabv3plus_r50-d8_512x512_80k_loveda.py' 2 | model = dict( 3 | backbone=dict( 4 | depth=101, 5 | init_cfg=dict( 6 | type='Pretrained', checkpoint='open-mmlab://resnet101_v1c'))) 7 | -------------------------------------------------------------------------------- /software/mmseg/configs/deeplabv3plus/deeplabv3plus_r101-d8_512x512_80k_potsdam.py: -------------------------------------------------------------------------------- 1 | _base_ = './deeplabv3plus_r50-d8_512x512_80k_potsdam.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/deeplabv3plus/deeplabv3plus_r101-d8_769x769_40k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './deeplabv3plus_r50-d8_769x769_40k_cityscapes.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/deeplabv3plus/deeplabv3plus_r101-d8_769x769_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './deeplabv3plus_r50-d8_769x769_80k_cityscapes.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/deeplabv3plus/deeplabv3plus_r101-d8_fp16_512x1024_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './deeplabv3plus_r101-d8_512x1024_80k_cityscapes.py' 2 | # fp16 settings 3 | optimizer_config = dict(type='Fp16OptimizerHook', loss_scale=512.) 4 | # fp16 placeholder 5 | fp16 = dict() 6 | -------------------------------------------------------------------------------- /software/mmseg/configs/deeplabv3plus/deeplabv3plus_r101b-d8_512x1024_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './deeplabv3plus_r50-d8_512x1024_80k_cityscapes.py' 2 | model = dict( 3 | pretrained='torchvision://resnet101', 4 | backbone=dict(type='ResNet', depth=101)) 5 | -------------------------------------------------------------------------------- /software/mmseg/configs/deeplabv3plus/deeplabv3plus_r101b-d8_769x769_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './deeplabv3plus_r50-d8_769x769_80k_cityscapes.py' 2 | model = dict( 3 | pretrained='torchvision://resnet101', 4 | backbone=dict(type='ResNet', depth=101)) 5 | -------------------------------------------------------------------------------- /software/mmseg/configs/deeplabv3plus/deeplabv3plus_r50-d8_512x1024_40k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/deeplabv3plus_r50-d8.py', 3 | '../_base_/datasets/cityscapes.py', '../_base_/default_runtime.py', 4 | '../_base_/schedules/schedule_40k.py' 5 | ] 6 | -------------------------------------------------------------------------------- /software/mmseg/configs/deeplabv3plus/deeplabv3plus_r50-d8_512x1024_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/deeplabv3plus_r50-d8.py', 3 | '../_base_/datasets/cityscapes.py', '../_base_/default_runtime.py', 4 | '../_base_/schedules/schedule_80k.py' 5 | ] 6 | -------------------------------------------------------------------------------- /software/mmseg/configs/deeplabv3plus/deeplabv3plus_r50b-d8_512x1024_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './deeplabv3plus_r50-d8_512x1024_80k_cityscapes.py' 2 | model = dict(pretrained='torchvision://resnet50', backbone=dict(type='ResNet')) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/deeplabv3plus/deeplabv3plus_r50b-d8_769x769_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './deeplabv3plus_r50-d8_769x769_80k_cityscapes.py' 2 | model = dict(pretrained='torchvision://resnet50', backbone=dict(type='ResNet')) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/dmnet/dmnet_r101-d8_512x1024_40k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './dmnet_r50-d8_512x1024_40k_cityscapes.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/dmnet/dmnet_r101-d8_512x1024_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './dmnet_r50-d8_512x1024_80k_cityscapes.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/dmnet/dmnet_r101-d8_512x512_160k_ade20k.py: -------------------------------------------------------------------------------- 1 | _base_ = './dmnet_r50-d8_512x512_160k_ade20k.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/dmnet/dmnet_r101-d8_512x512_80k_ade20k.py: -------------------------------------------------------------------------------- 1 | _base_ = './dmnet_r50-d8_512x512_80k_ade20k.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/dmnet/dmnet_r101-d8_769x769_40k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './dmnet_r50-d8_769x769_40k_cityscapes.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/dmnet/dmnet_r101-d8_769x769_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './dmnet_r50-d8_769x769_80k_cityscapes.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/dmnet/dmnet_r50-d8_512x1024_40k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/dmnet_r50-d8.py', '../_base_/datasets/cityscapes.py', 3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_40k.py' 4 | ] 5 | -------------------------------------------------------------------------------- /software/mmseg/configs/dmnet/dmnet_r50-d8_512x1024_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/dmnet_r50-d8.py', '../_base_/datasets/cityscapes.py', 3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_80k.py' 4 | ] 5 | -------------------------------------------------------------------------------- /software/mmseg/configs/dnlnet/dnl_r101-d8_512x1024_40k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './dnl_r50-d8_512x1024_40k_cityscapes.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/dnlnet/dnl_r101-d8_512x1024_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './dnl_r50-d8_512x1024_80k_cityscapes.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/dnlnet/dnl_r101-d8_512x512_160k_ade20k.py: -------------------------------------------------------------------------------- 1 | _base_ = './dnl_r50-d8_512x512_160k_ade20k.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/dnlnet/dnl_r101-d8_512x512_80k_ade20k.py: -------------------------------------------------------------------------------- 1 | _base_ = './dnl_r50-d8_512x512_80k_ade20k.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/dnlnet/dnl_r101-d8_769x769_40k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './dnl_r50-d8_769x769_40k_cityscapes.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/dnlnet/dnl_r101-d8_769x769_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './dnl_r50-d8_769x769_80k_cityscapes.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/dnlnet/dnl_r50-d8_512x1024_40k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/dnl_r50-d8.py', '../_base_/datasets/cityscapes.py', 3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_40k.py' 4 | ] 5 | -------------------------------------------------------------------------------- /software/mmseg/configs/dnlnet/dnl_r50-d8_512x1024_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/dnl_r50-d8.py', '../_base_/datasets/cityscapes.py', 3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_80k.py' 4 | ] 5 | -------------------------------------------------------------------------------- /software/mmseg/configs/emanet/emanet_r101-d8_512x1024_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './emanet_r50-d8_512x1024_80k_cityscapes.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/emanet/emanet_r101-d8_769x769_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './emanet_r50-d8_769x769_80k_cityscapes.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/emanet/emanet_r50-d8_512x1024_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/emanet_r50-d8.py', '../_base_/datasets/cityscapes.py', 3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_80k.py' 4 | ] 5 | -------------------------------------------------------------------------------- /software/mmseg/configs/encnet/encnet_r101-d8_512x1024_40k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './encnet_r50-d8_512x1024_40k_cityscapes.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/encnet/encnet_r101-d8_512x1024_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './encnet_r50-d8_512x1024_80k_cityscapes.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/encnet/encnet_r101-d8_512x512_160k_ade20k.py: -------------------------------------------------------------------------------- 1 | _base_ = './encnet_r50-d8_512x512_160k_ade20k.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/encnet/encnet_r101-d8_512x512_20k_voc12aug.py: -------------------------------------------------------------------------------- 1 | _base_ = './encnet_r50-d8_512x512_20k_voc12aug.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/encnet/encnet_r101-d8_512x512_40k_voc12aug.py: -------------------------------------------------------------------------------- 1 | _base_ = './encnet_r50-d8_512x512_40k_voc12aug.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/encnet/encnet_r101-d8_512x512_80k_ade20k.py: -------------------------------------------------------------------------------- 1 | _base_ = './encnet_r50-d8_512x512_80k_ade20k.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/encnet/encnet_r101-d8_769x769_40k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './encnet_r50-d8_769x769_40k_cityscapes.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/encnet/encnet_r101-d8_769x769_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './encnet_r50-d8_769x769_80k_cityscapes.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/encnet/encnet_r50-d8_512x1024_40k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/encnet_r50-d8.py', '../_base_/datasets/cityscapes.py', 3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_40k.py' 4 | ] 5 | -------------------------------------------------------------------------------- /software/mmseg/configs/encnet/encnet_r50-d8_512x1024_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/encnet_r50-d8.py', '../_base_/datasets/cityscapes.py', 3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_80k.py' 4 | ] 5 | -------------------------------------------------------------------------------- /software/mmseg/configs/erfnet/erfnet_fcn_4x4_512x1024_160k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/erfnet_fcn.py', '../_base_/datasets/cityscapes.py', 3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_160k.py' 4 | ] 5 | data = dict( 6 | samples_per_gpu=4, 7 | workers_per_gpu=4, 8 | ) 9 | -------------------------------------------------------------------------------- /software/mmseg/configs/fastfcn/fastfcn_r50-d32_jpu_aspp_4x4_512x1024_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | # model settings 2 | _base_ = './fastfcn_r50-d32_jpu_aspp_512x1024_80k_cityscapes.py' 3 | data = dict( 4 | samples_per_gpu=4, 5 | workers_per_gpu=4, 6 | ) 7 | -------------------------------------------------------------------------------- /software/mmseg/configs/fastfcn/fastfcn_r50-d32_jpu_enc_4x4_512x1024_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | # model settings 2 | _base_ = './fastfcn_r50-d32_jpu_enc_512x1024_80k_cityscapes.py' 3 | data = dict( 4 | samples_per_gpu=4, 5 | workers_per_gpu=4, 6 | ) 7 | -------------------------------------------------------------------------------- /software/mmseg/configs/fastfcn/fastfcn_r50-d32_jpu_psp_512x1024_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/fastfcn_r50-d32_jpu_psp.py', 3 | '../_base_/datasets/cityscapes.py', '../_base_/default_runtime.py', 4 | '../_base_/schedules/schedule_80k.py' 5 | ] 6 | -------------------------------------------------------------------------------- /software/mmseg/configs/fcn/fcn_d6_r101-d16_512x1024_40k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './fcn_d6_r50-d16_512x1024_40k_cityscapes.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/fcn/fcn_d6_r101-d16_512x1024_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './fcn_d6_r50-d16_512x1024_80k_cityscapes.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/fcn/fcn_d6_r101-d16_769x769_40k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './fcn_d6_r50-d16_769x769_40k_cityscapes.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/fcn/fcn_d6_r101-d16_769x769_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './fcn_d6_r50-d16_769x769_80k_cityscapes.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/fcn/fcn_d6_r101b-d16_512x1024_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './fcn_d6_r50b-d16_512x1024_80k_cityscapes.py' 2 | model = dict( 3 | pretrained='torchvision://resnet101', 4 | backbone=dict(type='ResNet', depth=101)) 5 | -------------------------------------------------------------------------------- /software/mmseg/configs/fcn/fcn_d6_r101b-d16_769x769_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './fcn_d6_r50b-d16_769x769_80k_cityscapes.py' 2 | model = dict( 3 | pretrained='torchvision://resnet101', 4 | backbone=dict(type='ResNet', depth=101)) 5 | -------------------------------------------------------------------------------- /software/mmseg/configs/fcn/fcn_d6_r50b-d16_512x1024_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './fcn_d6_r50-d16_512x1024_80k_cityscapes.py' 2 | model = dict(pretrained='torchvision://resnet50', backbone=dict(type='ResNet')) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/fcn/fcn_d6_r50b-d16_769x769_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './fcn_d6_r50-d16_769x769_80k_cityscapes.py' 2 | model = dict(pretrained='torchvision://resnet50', backbone=dict(type='ResNet')) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/fcn/fcn_r101-d8_480x480_40k_pascal_context.py: -------------------------------------------------------------------------------- 1 | _base_ = './fcn_r50-d8_480x480_40k_pascal_context.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/fcn/fcn_r101-d8_480x480_40k_pascal_context_59.py: -------------------------------------------------------------------------------- 1 | _base_ = './fcn_r50-d8_480x480_40k_pascal_context_59.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/fcn/fcn_r101-d8_480x480_80k_pascal_context.py: -------------------------------------------------------------------------------- 1 | _base_ = './fcn_r50-d8_480x480_80k_pascal_context.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/fcn/fcn_r101-d8_480x480_80k_pascal_context_59.py: -------------------------------------------------------------------------------- 1 | _base_ = './fcn_r50-d8_480x480_80k_pascal_context_59.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/fcn/fcn_r101-d8_512x1024_40k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './fcn_r50-d8_512x1024_40k_cityscapes.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/fcn/fcn_r101-d8_512x1024_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './fcn_r50-d8_512x1024_80k_cityscapes.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/fcn/fcn_r101-d8_512x512_160k_ade20k.py: -------------------------------------------------------------------------------- 1 | _base_ = './fcn_r50-d8_512x512_160k_ade20k.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/fcn/fcn_r101-d8_512x512_20k_voc12aug.py: -------------------------------------------------------------------------------- 1 | _base_ = './fcn_r50-d8_512x512_20k_voc12aug.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/fcn/fcn_r101-d8_512x512_40k_voc12aug.py: -------------------------------------------------------------------------------- 1 | _base_ = './fcn_r50-d8_512x512_40k_voc12aug.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/fcn/fcn_r101-d8_512x512_80k_ade20k.py: -------------------------------------------------------------------------------- 1 | _base_ = './fcn_r50-d8_512x512_80k_ade20k.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/fcn/fcn_r101-d8_769x769_40k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './fcn_r50-d8_769x769_40k_cityscapes.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/fcn/fcn_r101-d8_769x769_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './fcn_r50-d8_769x769_80k_cityscapes.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/fcn/fcn_r101-d8_fp16_512x1024_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './fcn_r101-d8_512x1024_80k_cityscapes.py' 2 | # fp16 settings 3 | optimizer_config = dict(type='Fp16OptimizerHook', loss_scale=512.) 4 | # fp16 placeholder 5 | fp16 = dict() 6 | -------------------------------------------------------------------------------- /software/mmseg/configs/fcn/fcn_r101b-d8_512x1024_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './fcn_r50-d8_512x1024_80k_cityscapes.py' 2 | model = dict( 3 | pretrained='torchvision://resnet101', 4 | backbone=dict(type='ResNet', depth=101)) 5 | -------------------------------------------------------------------------------- /software/mmseg/configs/fcn/fcn_r101b-d8_769x769_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './fcn_r50-d8_769x769_80k_cityscapes.py' 2 | model = dict( 3 | pretrained='torchvision://resnet101', 4 | backbone=dict(type='ResNet', depth=101)) 5 | -------------------------------------------------------------------------------- /software/mmseg/configs/fcn/fcn_r50-d8_512x1024_40k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/fcn_r50-d8.py', '../_base_/datasets/cityscapes.py', 3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_40k.py' 4 | ] 5 | -------------------------------------------------------------------------------- /software/mmseg/configs/fcn/fcn_r50-d8_512x1024_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/fcn_r50-d8.py', '../_base_/datasets/cityscapes.py', 3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_80k.py' 4 | ] 5 | -------------------------------------------------------------------------------- /software/mmseg/configs/fcn/fcn_r50b-d8_512x1024_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './fcn_r50-d8_512x1024_80k_cityscapes.py' 2 | model = dict(pretrained='torchvision://resnet50', backbone=dict(type='ResNet')) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/fcn/fcn_r50b-d8_769x769_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './fcn_r50-d8_769x769_80k_cityscapes.py' 2 | model = dict(pretrained='torchvision://resnet50', backbone=dict(type='ResNet')) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/gcnet/gcnet_r101-d8_512x1024_40k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './gcnet_r50-d8_512x1024_40k_cityscapes.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/gcnet/gcnet_r101-d8_512x1024_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './gcnet_r50-d8_512x1024_80k_cityscapes.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/gcnet/gcnet_r101-d8_512x512_160k_ade20k.py: -------------------------------------------------------------------------------- 1 | _base_ = './gcnet_r50-d8_512x512_160k_ade20k.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/gcnet/gcnet_r101-d8_512x512_20k_voc12aug.py: -------------------------------------------------------------------------------- 1 | _base_ = './gcnet_r50-d8_512x512_20k_voc12aug.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/gcnet/gcnet_r101-d8_512x512_40k_voc12aug.py: -------------------------------------------------------------------------------- 1 | _base_ = './gcnet_r50-d8_512x512_40k_voc12aug.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/gcnet/gcnet_r101-d8_512x512_80k_ade20k.py: -------------------------------------------------------------------------------- 1 | _base_ = './gcnet_r50-d8_512x512_80k_ade20k.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/gcnet/gcnet_r101-d8_769x769_40k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './gcnet_r50-d8_769x769_40k_cityscapes.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/gcnet/gcnet_r101-d8_769x769_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './gcnet_r50-d8_769x769_80k_cityscapes.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/gcnet/gcnet_r50-d8_512x1024_40k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/gcnet_r50-d8.py', '../_base_/datasets/cityscapes.py', 3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_40k.py' 4 | ] 5 | -------------------------------------------------------------------------------- /software/mmseg/configs/gcnet/gcnet_r50-d8_512x1024_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/gcnet_r50-d8.py', '../_base_/datasets/cityscapes.py', 3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_80k.py' 4 | ] 5 | -------------------------------------------------------------------------------- /software/mmseg/configs/hrnet/fcn_hr18_4x4_512x512_80k_vaihingen.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/fcn_hr18.py', '../_base_/datasets/vaihingen.py', 3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_80k.py' 4 | ] 5 | model = dict(decode_head=dict(num_classes=6)) 6 | -------------------------------------------------------------------------------- /software/mmseg/configs/hrnet/fcn_hr18_4x4_896x896_80k_isaid.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/fcn_hr18.py', '../_base_/datasets/isaid.py', 3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_80k.py' 4 | ] 5 | model = dict(decode_head=dict(num_classes=16)) 6 | -------------------------------------------------------------------------------- /software/mmseg/configs/hrnet/fcn_hr18_512x1024_160k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/fcn_hr18.py', '../_base_/datasets/cityscapes.py', 3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_160k.py' 4 | ] 5 | -------------------------------------------------------------------------------- /software/mmseg/configs/hrnet/fcn_hr18_512x1024_40k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/fcn_hr18.py', '../_base_/datasets/cityscapes.py', 3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_40k.py' 4 | ] 5 | -------------------------------------------------------------------------------- /software/mmseg/configs/hrnet/fcn_hr18_512x1024_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/fcn_hr18.py', '../_base_/datasets/cityscapes.py', 3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_80k.py' 4 | ] 5 | -------------------------------------------------------------------------------- /software/mmseg/configs/hrnet/fcn_hr18_512x512_160k_ade20k.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/fcn_hr18.py', '../_base_/datasets/ade20k.py', 3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_160k.py' 4 | ] 5 | model = dict(decode_head=dict(num_classes=150)) 6 | -------------------------------------------------------------------------------- /software/mmseg/configs/hrnet/fcn_hr18_512x512_20k_voc12aug.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/fcn_hr18.py', '../_base_/datasets/pascal_voc12_aug.py', 3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_20k.py' 4 | ] 5 | model = dict(decode_head=dict(num_classes=21)) 6 | -------------------------------------------------------------------------------- /software/mmseg/configs/hrnet/fcn_hr18_512x512_40k_voc12aug.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/fcn_hr18.py', '../_base_/datasets/pascal_voc12_aug.py', 3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_40k.py' 4 | ] 5 | model = dict(decode_head=dict(num_classes=21)) 6 | -------------------------------------------------------------------------------- /software/mmseg/configs/hrnet/fcn_hr18_512x512_80k_ade20k.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/fcn_hr18.py', '../_base_/datasets/ade20k.py', 3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_80k.py' 4 | ] 5 | model = dict(decode_head=dict(num_classes=150)) 6 | -------------------------------------------------------------------------------- /software/mmseg/configs/hrnet/fcn_hr18_512x512_80k_loveda.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/fcn_hr18.py', '../_base_/datasets/loveda.py', 3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_80k.py' 4 | ] 5 | model = dict(decode_head=dict(num_classes=7)) 6 | -------------------------------------------------------------------------------- /software/mmseg/configs/hrnet/fcn_hr18_512x512_80k_potsdam.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/fcn_hr18.py', '../_base_/datasets/potsdam.py', 3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_80k.py' 4 | ] 5 | model = dict(decode_head=dict(num_classes=6)) 6 | -------------------------------------------------------------------------------- /software/mmseg/configs/icnet/icnet_r101-d8_832x832_160k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './icnet_r50-d8_832x832_160k_cityscapes.py' 2 | model = dict(backbone=dict(backbone_cfg=dict(depth=101))) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/icnet/icnet_r101-d8_832x832_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './icnet_r50-d8_832x832_80k_cityscapes.py' 2 | model = dict(backbone=dict(backbone_cfg=dict(depth=101))) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/icnet/icnet_r18-d8_832x832_160k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './icnet_r50-d8_832x832_160k_cityscapes.py' 2 | model = dict( 3 | backbone=dict(layer_channels=(128, 512), backbone_cfg=dict(depth=18))) 4 | -------------------------------------------------------------------------------- /software/mmseg/configs/icnet/icnet_r18-d8_832x832_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './icnet_r50-d8_832x832_80k_cityscapes.py' 2 | model = dict( 3 | backbone=dict(layer_channels=(128, 512), backbone_cfg=dict(depth=18))) 4 | -------------------------------------------------------------------------------- /software/mmseg/configs/icnet/icnet_r50-d8_832x832_160k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/icnet_r50-d8.py', 3 | '../_base_/datasets/cityscapes_832x832.py', '../_base_/default_runtime.py', 4 | '../_base_/schedules/schedule_160k.py' 5 | ] 6 | -------------------------------------------------------------------------------- /software/mmseg/configs/icnet/icnet_r50-d8_832x832_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/icnet_r50-d8.py', 3 | '../_base_/datasets/cityscapes_832x832.py', '../_base_/default_runtime.py', 4 | '../_base_/schedules/schedule_80k.py' 5 | ] 6 | -------------------------------------------------------------------------------- /software/mmseg/configs/icnet/icnet_r50-d8_in1k-pre_832x832_160k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './icnet_r50-d8_832x832_160k_cityscapes.py' 2 | model = dict( 3 | backbone=dict( 4 | backbone_cfg=dict( 5 | init_cfg=dict( 6 | type='Pretrained', checkpoint='open-mmlab://resnet50_v1c')))) 7 | -------------------------------------------------------------------------------- /software/mmseg/configs/icnet/icnet_r50-d8_in1k-pre_832x832_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './icnet_r50-d8_832x832_80k_cityscapes.py' 2 | model = dict( 3 | backbone=dict( 4 | backbone_cfg=dict( 5 | init_cfg=dict( 6 | type='Pretrained', checkpoint='open-mmlab://resnet50_v1c')))) 7 | -------------------------------------------------------------------------------- /software/mmseg/configs/isanet/isanet_r101-d8_512x1024_40k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './isanet_r50-d8_512x1024_40k_cityscapes.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/isanet/isanet_r101-d8_512x1024_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './isanet_r50-d8_512x1024_80k_cityscapes.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/isanet/isanet_r101-d8_512x512_160k_ade20k.py: -------------------------------------------------------------------------------- 1 | _base_ = './isanet_r50-d8_512x512_160k_ade20k.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/isanet/isanet_r101-d8_512x512_20k_voc12aug.py: -------------------------------------------------------------------------------- 1 | _base_ = './isanet_r50-d8_512x512_20k_voc12aug.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/isanet/isanet_r101-d8_512x512_40k_voc12aug.py: -------------------------------------------------------------------------------- 1 | _base_ = './isanet_r50-d8_512x512_40k_voc12aug.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/isanet/isanet_r101-d8_512x512_80k_ade20k.py: -------------------------------------------------------------------------------- 1 | _base_ = './isanet_r50-d8_512x512_80k_ade20k.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/isanet/isanet_r101-d8_769x769_40k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './isanet_r50-d8_769x769_40k_cityscapes.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/isanet/isanet_r101-d8_769x769_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './isanet_r50-d8_769x769_80k_cityscapes.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/isanet/isanet_r50-d8_512x1024_40k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/isanet_r50-d8.py', '../_base_/datasets/cityscapes.py', 3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_40k.py' 4 | ] 5 | -------------------------------------------------------------------------------- /software/mmseg/configs/isanet/isanet_r50-d8_512x1024_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/isanet_r50-d8.py', '../_base_/datasets/cityscapes.py', 3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_80k.py' 4 | ] 5 | -------------------------------------------------------------------------------- /software/mmseg/configs/nonlocal_net/nonlocal_r101-d8_512x1024_40k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './nonlocal_r50-d8_512x1024_40k_cityscapes.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/nonlocal_net/nonlocal_r101-d8_512x1024_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './nonlocal_r50-d8_512x1024_80k_cityscapes.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/nonlocal_net/nonlocal_r101-d8_512x512_160k_ade20k.py: -------------------------------------------------------------------------------- 1 | _base_ = './nonlocal_r50-d8_512x512_160k_ade20k.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/nonlocal_net/nonlocal_r101-d8_512x512_20k_voc12aug.py: -------------------------------------------------------------------------------- 1 | _base_ = './nonlocal_r50-d8_512x512_20k_voc12aug.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/nonlocal_net/nonlocal_r101-d8_512x512_40k_voc12aug.py: -------------------------------------------------------------------------------- 1 | _base_ = './nonlocal_r50-d8_512x512_40k_voc12aug.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/nonlocal_net/nonlocal_r101-d8_512x512_80k_ade20k.py: -------------------------------------------------------------------------------- 1 | _base_ = './nonlocal_r50-d8_512x512_80k_ade20k.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/nonlocal_net/nonlocal_r101-d8_769x769_40k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './nonlocal_r50-d8_769x769_40k_cityscapes.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/nonlocal_net/nonlocal_r101-d8_769x769_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './nonlocal_r50-d8_769x769_80k_cityscapes.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/nonlocal_net/nonlocal_r50-d8_512x1024_40k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/nonlocal_r50-d8.py', '../_base_/datasets/cityscapes.py', 3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_40k.py' 4 | ] 5 | -------------------------------------------------------------------------------- /software/mmseg/configs/nonlocal_net/nonlocal_r50-d8_512x1024_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/nonlocal_r50-d8.py', '../_base_/datasets/cityscapes.py', 3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_80k.py' 4 | ] 5 | -------------------------------------------------------------------------------- /software/mmseg/configs/ocrnet/ocrnet_hr18_512x1024_160k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/ocrnet_hr18.py', '../_base_/datasets/cityscapes.py', 3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_160k.py' 4 | ] 5 | -------------------------------------------------------------------------------- /software/mmseg/configs/ocrnet/ocrnet_hr18_512x1024_40k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/ocrnet_hr18.py', '../_base_/datasets/cityscapes.py', 3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_40k.py' 4 | ] 5 | -------------------------------------------------------------------------------- /software/mmseg/configs/ocrnet/ocrnet_hr18_512x1024_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/ocrnet_hr18.py', '../_base_/datasets/cityscapes.py', 3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_80k.py' 4 | ] 5 | -------------------------------------------------------------------------------- /software/mmseg/configs/point_rend/pointrend_r101_512x1024_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './pointrend_r50_512x1024_80k_cityscapes.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/point_rend/pointrend_r101_512x512_160k_ade20k.py: -------------------------------------------------------------------------------- 1 | _base_ = './pointrend_r50_512x512_160k_ade20k.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/point_rend/pointrend_r50_512x1024_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/pointrend_r50.py', '../_base_/datasets/cityscapes.py', 3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_80k.py' 4 | ] 5 | lr_config = dict(warmup='linear', warmup_iters=200) 6 | -------------------------------------------------------------------------------- /software/mmseg/configs/psanet/psanet_r101-d8_512x1024_40k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './psanet_r50-d8_512x1024_40k_cityscapes.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/psanet/psanet_r101-d8_512x1024_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './psanet_r50-d8_512x1024_80k_cityscapes.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/psanet/psanet_r101-d8_512x512_160k_ade20k.py: -------------------------------------------------------------------------------- 1 | _base_ = './psanet_r50-d8_512x512_160k_ade20k.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/psanet/psanet_r101-d8_512x512_20k_voc12aug.py: -------------------------------------------------------------------------------- 1 | _base_ = './psanet_r50-d8_512x512_20k_voc12aug.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/psanet/psanet_r101-d8_512x512_40k_voc12aug.py: -------------------------------------------------------------------------------- 1 | _base_ = './psanet_r50-d8_512x512_40k_voc12aug.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/psanet/psanet_r101-d8_512x512_80k_ade20k.py: -------------------------------------------------------------------------------- 1 | _base_ = './psanet_r50-d8_512x512_80k_ade20k.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/psanet/psanet_r101-d8_769x769_40k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './psanet_r50-d8_769x769_40k_cityscapes.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/psanet/psanet_r101-d8_769x769_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './psanet_r50-d8_769x769_80k_cityscapes.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/psanet/psanet_r50-d8_512x1024_40k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/psanet_r50-d8.py', '../_base_/datasets/cityscapes.py', 3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_40k.py' 4 | ] 5 | -------------------------------------------------------------------------------- /software/mmseg/configs/psanet/psanet_r50-d8_512x1024_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/psanet_r50-d8.py', '../_base_/datasets/cityscapes.py', 3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_80k.py' 4 | ] 5 | -------------------------------------------------------------------------------- /software/mmseg/configs/pspnet/pspnet_r101-d8_480x480_40k_pascal_context.py: -------------------------------------------------------------------------------- 1 | _base_ = './pspnet_r50-d8_480x480_40k_pascal_context.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/pspnet/pspnet_r101-d8_480x480_40k_pascal_context_59.py: -------------------------------------------------------------------------------- 1 | _base_ = './pspnet_r50-d8_480x480_40k_pascal_context_59.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/pspnet/pspnet_r101-d8_480x480_80k_pascal_context.py: -------------------------------------------------------------------------------- 1 | _base_ = './pspnet_r50-d8_480x480_80k_pascal_context.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/pspnet/pspnet_r101-d8_480x480_80k_pascal_context_59.py: -------------------------------------------------------------------------------- 1 | _base_ = './pspnet_r50-d8_480x480_80k_pascal_context_59.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/pspnet/pspnet_r101-d8_4x4_512x512_80k_potsdam.py: -------------------------------------------------------------------------------- 1 | _base_ = './pspnet_r50-d8_4x4_512x512_80k_potsdam.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/pspnet/pspnet_r101-d8_4x4_512x512_80k_vaihingen.py: -------------------------------------------------------------------------------- 1 | _base_ = './pspnet_r50-d8_4x4_512x512_80k_vaihingen.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/pspnet/pspnet_r101-d8_512x1024_40k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './pspnet_r50-d8_512x1024_40k_cityscapes.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/pspnet/pspnet_r101-d8_512x1024_40k_dark.py: -------------------------------------------------------------------------------- 1 | _base_ = './pspnet_r50-d8_512x1024_40k_dark.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/pspnet/pspnet_r101-d8_512x1024_40k_night_driving.py: -------------------------------------------------------------------------------- 1 | _base_ = './pspnet_r50-d8_512x1024_40k_night_driving.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/pspnet/pspnet_r101-d8_512x1024_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './pspnet_r50-d8_512x1024_80k_cityscapes.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/pspnet/pspnet_r101-d8_512x512_160k_ade20k.py: -------------------------------------------------------------------------------- 1 | _base_ = './pspnet_r50-d8_512x512_160k_ade20k.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/pspnet/pspnet_r101-d8_512x512_20k_voc12aug.py: -------------------------------------------------------------------------------- 1 | _base_ = './pspnet_r50-d8_512x512_20k_voc12aug.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/pspnet/pspnet_r101-d8_512x512_40k_voc12aug.py: -------------------------------------------------------------------------------- 1 | _base_ = './pspnet_r50-d8_512x512_40k_voc12aug.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/pspnet/pspnet_r101-d8_512x512_4x4_160k_coco-stuff164k.py: -------------------------------------------------------------------------------- 1 | _base_ = './pspnet_r50-d8_512x512_4x4_160k_coco-stuff164k.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/pspnet/pspnet_r101-d8_512x512_4x4_20k_coco-stuff10k.py: -------------------------------------------------------------------------------- 1 | _base_ = './pspnet_r50-d8_512x512_4x4_20k_coco-stuff10k.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/pspnet/pspnet_r101-d8_512x512_4x4_320k_coco-stuff164k.py: -------------------------------------------------------------------------------- 1 | _base_ = './pspnet_r50-d8_512x512_4x4_320k_coco-stuff164k.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/pspnet/pspnet_r101-d8_512x512_4x4_40k_coco-stuff10k.py: -------------------------------------------------------------------------------- 1 | _base_ = './pspnet_r50-d8_512x512_4x4_40k_coco-stuff10k.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/pspnet/pspnet_r101-d8_512x512_4x4_80k_coco-stuff164k.py: -------------------------------------------------------------------------------- 1 | _base_ = './pspnet_r50-d8_512x512_4x4_80k_coco-stuff164k.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/pspnet/pspnet_r101-d8_512x512_80k_ade20k.py: -------------------------------------------------------------------------------- 1 | _base_ = './pspnet_r50-d8_512x512_80k_ade20k.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/pspnet/pspnet_r101-d8_512x512_80k_loveda.py: -------------------------------------------------------------------------------- 1 | _base_ = './pspnet_r50-d8_512x512_80k_loveda.py' 2 | model = dict( 3 | backbone=dict( 4 | depth=101, 5 | init_cfg=dict( 6 | type='Pretrained', checkpoint='open-mmlab://resnet101_v1c'))) 7 | -------------------------------------------------------------------------------- /software/mmseg/configs/pspnet/pspnet_r101-d8_769x769_40k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './pspnet_r50-d8_769x769_40k_cityscapes.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/pspnet/pspnet_r101-d8_769x769_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './pspnet_r50-d8_769x769_80k_cityscapes.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/pspnet/pspnet_r101-d8_fp16_512x1024_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './pspnet_r101-d8_512x1024_80k_cityscapes.py' 2 | # fp16 settings 3 | optimizer_config = dict(type='Fp16OptimizerHook', loss_scale=512.) 4 | # fp16 placeholder 5 | fp16 = dict() 6 | -------------------------------------------------------------------------------- /software/mmseg/configs/pspnet/pspnet_r101b-d8_512x1024_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './pspnet_r50-d8_512x1024_80k_cityscapes.py' 2 | model = dict( 3 | pretrained='torchvision://resnet101', 4 | backbone=dict(type='ResNet', depth=101)) 5 | -------------------------------------------------------------------------------- /software/mmseg/configs/pspnet/pspnet_r101b-d8_512x1024_80k_dark.py: -------------------------------------------------------------------------------- 1 | _base_ = './pspnet_r50-d8_512x1024_80k_dark.py' 2 | model = dict( 3 | pretrained='torchvision://resnet101', 4 | backbone=dict(type='ResNet', depth=101)) 5 | -------------------------------------------------------------------------------- /software/mmseg/configs/pspnet/pspnet_r101b-d8_512x1024_80k_night_driving.py: -------------------------------------------------------------------------------- 1 | _base_ = './pspnet_r50-d8_512x1024_80k_night_driving.py' 2 | model = dict( 3 | pretrained='torchvision://resnet101', 4 | backbone=dict(type='ResNet', depth=101)) 5 | -------------------------------------------------------------------------------- /software/mmseg/configs/pspnet/pspnet_r101b-d8_769x769_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './pspnet_r50-d8_769x769_80k_cityscapes.py' 2 | model = dict( 3 | pretrained='torchvision://resnet101', 4 | backbone=dict(type='ResNet', depth=101)) 5 | -------------------------------------------------------------------------------- /software/mmseg/configs/pspnet/pspnet_r50-d8_512x1024_40k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/pspnet_r50-d8.py', '../_base_/datasets/cityscapes.py', 3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_40k.py' 4 | ] 5 | -------------------------------------------------------------------------------- /software/mmseg/configs/pspnet/pspnet_r50-d8_512x1024_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/pspnet_r50-d8.py', '../_base_/datasets/cityscapes.py', 3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_80k.py' 4 | ] 5 | -------------------------------------------------------------------------------- /software/mmseg/configs/pspnet/pspnet_r50b-d8_512x1024_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './pspnet_r50-d8_512x1024_80k_cityscapes.py' 2 | model = dict(pretrained='torchvision://resnet50', backbone=dict(type='ResNet')) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/pspnet/pspnet_r50b-d8_769x769_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './pspnet_r50-d8_769x769_80k_cityscapes.py' 2 | model = dict(pretrained='torchvision://resnet50', backbone=dict(type='ResNet')) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/sem_fpn/fpn_r101_512x1024_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './fpn_r50_512x1024_80k_cityscapes.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/sem_fpn/fpn_r101_512x512_160k_ade20k.py: -------------------------------------------------------------------------------- 1 | _base_ = './fpn_r50_512x512_160k_ade20k.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/sem_fpn/fpn_r50_512x1024_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/fpn_r50.py', '../_base_/datasets/cityscapes.py', 3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_80k.py' 4 | ] 5 | -------------------------------------------------------------------------------- /software/mmseg/configs/sem_fpn/fpn_r50_512x512_160k_ade20k.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/fpn_r50.py', '../_base_/datasets/ade20k.py', 3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_160k.py' 4 | ] 5 | model = dict(decode_head=dict(num_classes=150)) 6 | -------------------------------------------------------------------------------- /software/mmseg/configs/setr/setr_mla_512x512_160k_b16_ade20k.py: -------------------------------------------------------------------------------- 1 | _base_ = ['./setr_mla_512x512_160k_b8_ade20k.py'] 2 | 3 | # num_gpus: 8 -> batch_size: 16 4 | data = dict(samples_per_gpu=2) 5 | -------------------------------------------------------------------------------- /software/mmseg/configs/stdc/stdc2_512x1024_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './stdc1_512x1024_80k_cityscapes.py' 2 | model = dict(backbone=dict(backbone_cfg=dict(stdc_type='STDCNet2'))) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/swin/upernet_swin_base_patch4_window12_512x512_160k_ade20k_pretrain_384x384_22K.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | './upernet_swin_base_patch4_window12_512x512_160k_ade20k_' 3 | 'pretrain_384x384_1K.py' 4 | ] 5 | model = dict(pretrained='pretrain/swin_base_patch4_window12_384_22k.pth') 6 | -------------------------------------------------------------------------------- /software/mmseg/configs/swin/upernet_swin_base_patch4_window7_512x512_160k_ade20k_pretrain_224x224_22K.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | './upernet_swin_base_patch4_window7_512x512_160k_ade20k_' 3 | 'pretrain_224x224_1K.py' 4 | ] 5 | model = dict(pretrained='pretrain/swin_base_patch4_window7_224_22k.pth') 6 | -------------------------------------------------------------------------------- /software/mmseg/configs/upernet/upernet_r101_512x1024_40k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './upernet_r50_512x1024_40k_cityscapes.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/upernet/upernet_r101_512x1024_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './upernet_r50_512x1024_80k_cityscapes.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/upernet/upernet_r101_512x512_160k_ade20k.py: -------------------------------------------------------------------------------- 1 | _base_ = './upernet_r50_512x512_160k_ade20k.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/upernet/upernet_r101_512x512_20k_voc12aug.py: -------------------------------------------------------------------------------- 1 | _base_ = './upernet_r50_512x512_20k_voc12aug.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/upernet/upernet_r101_512x512_40k_voc12aug.py: -------------------------------------------------------------------------------- 1 | _base_ = './upernet_r50_512x512_40k_voc12aug.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/upernet/upernet_r101_512x512_80k_ade20k.py: -------------------------------------------------------------------------------- 1 | _base_ = './upernet_r50_512x512_80k_ade20k.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/upernet/upernet_r101_769x769_40k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './upernet_r50_769x769_40k_cityscapes.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/upernet/upernet_r101_769x769_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = './upernet_r50_769x769_80k_cityscapes.py' 2 | model = dict(pretrained='open-mmlab://resnet101_v1c', backbone=dict(depth=101)) 3 | -------------------------------------------------------------------------------- /software/mmseg/configs/upernet/upernet_r50_512x1024_40k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/upernet_r50.py', '../_base_/datasets/cityscapes.py', 3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_40k.py' 4 | ] 5 | -------------------------------------------------------------------------------- /software/mmseg/configs/upernet/upernet_r50_512x1024_80k_cityscapes.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/upernet_r50.py', '../_base_/datasets/cityscapes.py', 3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_80k.py' 4 | ] 5 | -------------------------------------------------------------------------------- /software/mmseg/configs/vit/upernet_deit-b16_512x512_160k_ade20k.py: -------------------------------------------------------------------------------- 1 | _base_ = './upernet_vit-b16_mln_512x512_160k_ade20k.py' 2 | 3 | model = dict( 4 | pretrained='pretrain/deit_base_patch16_224-b5f2ef4d.pth', 5 | backbone=dict(drop_path_rate=0.1), 6 | neck=None) 7 | -------------------------------------------------------------------------------- /software/mmseg/configs/vit/upernet_deit-b16_512x512_80k_ade20k.py: -------------------------------------------------------------------------------- 1 | _base_ = './upernet_vit-b16_mln_512x512_80k_ade20k.py' 2 | 3 | model = dict( 4 | pretrained='pretrain/deit_base_patch16_224-b5f2ef4d.pth', 5 | backbone=dict(drop_path_rate=0.1), 6 | neck=None) 7 | -------------------------------------------------------------------------------- /software/mmseg/configs/vit/upernet_deit-b16_ln_mln_512x512_160k_ade20k.py: -------------------------------------------------------------------------------- 1 | _base_ = './upernet_vit-b16_mln_512x512_160k_ade20k.py' 2 | 3 | model = dict( 4 | pretrained='pretrain/deit_base_patch16_224-b5f2ef4d.pth', 5 | backbone=dict(drop_path_rate=0.1, final_norm=True)) 6 | -------------------------------------------------------------------------------- /software/mmseg/configs/vit/upernet_deit-b16_mln_512x512_160k_ade20k.py: -------------------------------------------------------------------------------- 1 | _base_ = './upernet_vit-b16_mln_512x512_160k_ade20k.py' 2 | 3 | model = dict( 4 | pretrained='pretrain/deit_base_patch16_224-b5f2ef4d.pth', 5 | backbone=dict(drop_path_rate=0.1), 6 | ) 7 | -------------------------------------------------------------------------------- /software/mmseg/docker/serve/config.properties: -------------------------------------------------------------------------------- 1 | inference_address=http://0.0.0.0:8080 2 | management_address=http://0.0.0.0:8081 3 | metrics_address=http://0.0.0.0:8082 4 | model_store=/home/model-server/model-store 5 | load_models=all 6 | -------------------------------------------------------------------------------- /software/mmseg/docker/serve/entrypoint.sh: -------------------------------------------------------------------------------- 1 | #!/bin/bash 2 | set -e 3 | 4 | if [[ "$1" = "serve" ]]; then 5 | shift 1 6 | torchserve --start --ts-config /home/model-server/config.properties 7 | else 8 | eval "$@" 9 | fi 10 | 11 | # prevent docker exit 12 | tail -f /dev/null 13 | -------------------------------------------------------------------------------- /software/mmseg/docs/en/_static/css/readthedocs.css: -------------------------------------------------------------------------------- 1 | .header-logo { 2 | background-image: url("../images/mmsegmentation.png"); 3 | background-size: 201px 40px; 4 | height: 40px; 5 | width: 201px; 6 | } 7 | -------------------------------------------------------------------------------- /software/mmseg/docs/en/switch_language.md: -------------------------------------------------------------------------------- 1 | ## English 2 | 3 | ## 简体中文 4 | -------------------------------------------------------------------------------- /software/mmseg/docs/en/tutorials/index.rst: -------------------------------------------------------------------------------- 1 | .. toctree:: 2 | :maxdepth: 2 3 | 4 | config.md 5 | customize_datasets.md 6 | data_pipeline.md 7 | customize_models.md 8 | training_tricks.md 9 | customize_runtime.md 10 | -------------------------------------------------------------------------------- /software/mmseg/docs/zh_cn/_static/css/readthedocs.css: -------------------------------------------------------------------------------- 1 | .header-logo { 2 | background-image: url("../images/mmsegmentation.png"); 3 | background-size: 201px 40px; 4 | height: 40px; 5 | width: 201px; 6 | } 7 | -------------------------------------------------------------------------------- /software/mmseg/docs/zh_cn/switch_language.md: -------------------------------------------------------------------------------- 1 | ## English 2 | 3 | ## 简体中文 4 | -------------------------------------------------------------------------------- /software/mmseg/docs/zh_cn/tutorials/index.rst: -------------------------------------------------------------------------------- 1 | .. toctree:: 2 | :maxdepth: 2 3 | 4 | config.md 5 | customize_datasets.md 6 | data_pipeline.md 7 | customize_models.md 8 | training_tricks.md 9 | customize_runtime.md 10 | -------------------------------------------------------------------------------- /software/mmseg/mmseg/.mim/configs: -------------------------------------------------------------------------------- 1 | ../../configs -------------------------------------------------------------------------------- /software/mmseg/mmseg/.mim/model-index.yml: -------------------------------------------------------------------------------- 1 | ../../model-index.yml -------------------------------------------------------------------------------- /software/mmseg/mmseg/.mim/tools: -------------------------------------------------------------------------------- 1 | ../../tools -------------------------------------------------------------------------------- /software/mmseg/mmseg/core/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) OpenMMLab. All rights reserved. 2 | from .evaluation import * # noqa: F401, F403 3 | from .seg import * # noqa: F401, F403 4 | from .utils import * # noqa: F401, F403 5 | -------------------------------------------------------------------------------- /software/mmseg/mmseg/core/seg/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) OpenMMLab. All rights reserved. 2 | from .builder import build_pixel_sampler 3 | from .sampler import BasePixelSampler, OHEMPixelSampler 4 | 5 | __all__ = ['build_pixel_sampler', 'BasePixelSampler', 'OHEMPixelSampler'] 6 | -------------------------------------------------------------------------------- /software/mmseg/mmseg/core/seg/sampler/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) OpenMMLab. All rights reserved. 2 | from .base_pixel_sampler import BasePixelSampler 3 | from .ohem_pixel_sampler import OHEMPixelSampler 4 | 5 | __all__ = ['BasePixelSampler', 'OHEMPixelSampler'] 6 | -------------------------------------------------------------------------------- /software/mmseg/mmseg/core/utils/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) OpenMMLab. All rights reserved. 2 | from .layer_decay_optimizer_constructor import \ 3 | LearningRateDecayOptimizerConstructor 4 | from .misc import add_prefix 5 | 6 | __all__ = ['add_prefix', 'LearningRateDecayOptimizerConstructor'] 7 | -------------------------------------------------------------------------------- /software/mmseg/mmseg/ops/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) OpenMMLab. All rights reserved. 2 | from .encoding import Encoding 3 | from .wrappers import Upsample, resize 4 | 5 | __all__ = ['Upsample', 'resize', 'Encoding'] 6 | -------------------------------------------------------------------------------- /software/mmseg/requirements.txt: -------------------------------------------------------------------------------- 1 | -r requirements/optional.txt 2 | -r requirements/runtime.txt 3 | -r requirements/tests.txt 4 | -------------------------------------------------------------------------------- /software/mmseg/requirements/docs.txt: -------------------------------------------------------------------------------- 1 | docutils==0.16.0 2 | myst-parser 3 | -e git+https://github.com/gaotongxiao/pytorch_sphinx_theme.git#egg=pytorch_sphinx_theme 4 | sphinx==4.0.2 5 | sphinx_copybutton 6 | sphinx_markdown_tables 7 | -------------------------------------------------------------------------------- /software/mmseg/requirements/mminstall.txt: -------------------------------------------------------------------------------- 1 | mmcls>=0.20.1 2 | mmcv-full>=1.4.4,<=1.5.0 3 | -------------------------------------------------------------------------------- /software/mmseg/requirements/optional.txt: -------------------------------------------------------------------------------- 1 | cityscapesscripts 2 | -------------------------------------------------------------------------------- /software/mmseg/requirements/readthedocs.txt: -------------------------------------------------------------------------------- 1 | mmcv 2 | prettytable 3 | torch 4 | torchvision 5 | -------------------------------------------------------------------------------- /software/mmseg/requirements/runtime.txt: -------------------------------------------------------------------------------- 1 | matplotlib 2 | mmcls>=0.20.1 3 | numpy 4 | packaging 5 | prettytable 6 | -------------------------------------------------------------------------------- /software/mmseg/requirements/tests.txt: -------------------------------------------------------------------------------- 1 | codecov 2 | flake8 3 | interrogate 4 | pytest 5 | xdoctest>=0.10.0 6 | yapf 7 | -------------------------------------------------------------------------------- /software/mmseg/resources/seg_demo.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenRobotLab/DORT/5c1f39259247813a5c340753dd1a274f253a39b7/software/mmseg/resources/seg_demo.gif -------------------------------------------------------------------------------- /software/mmseg/tests/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) OpenMMLab. All rights reserved. 2 | -------------------------------------------------------------------------------- /software/mmseg/tests/test_models/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) OpenMMLab. All rights reserved. 2 | -------------------------------------------------------------------------------- /software/mmseg/tests/test_models/test_backbones/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) OpenMMLab. All rights reserved. 2 | from .utils import all_zeros, check_norm_state, is_block, is_norm 3 | 4 | __all__ = ['is_norm', 'is_block', 'all_zeros', 'check_norm_state'] 5 | -------------------------------------------------------------------------------- /software/mmseg/tests/test_models/test_heads/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) OpenMMLab. All rights reserved. 2 | -------------------------------------------------------------------------------- /software/mmseg/tests/test_models/test_losses/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) OpenMMLab. All rights reserved. 2 | -------------------------------------------------------------------------------- /software/mmseg/tests/test_models/test_necks/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) OpenMMLab. All rights reserved. 2 | -------------------------------------------------------------------------------- /software/mmseg/tests/test_models/test_segmentors/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) OpenMMLab. All rights reserved. 2 | -------------------------------------------------------------------------------- /software/mmseg/tests/test_models/test_utils/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) OpenMMLab. All rights reserved. 2 | -------------------------------------------------------------------------------- /software/mmseg/tools/dist_train.sh: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env bash 2 | 3 | CONFIG=$1 4 | GPUS=$2 5 | PORT=${PORT:-29500} 6 | 7 | PYTHONPATH="$(dirname $0)/..":$PYTHONPATH \ 8 | python -m torch.distributed.launch --nproc_per_node=$GPUS --master_port=$PORT \ 9 | $(dirname "$0")/train.py $CONFIG --launcher pytorch ${@:3} 10 | -------------------------------------------------------------------------------- /software/mmtracking/MANIFEST.in: -------------------------------------------------------------------------------- 1 | include requirements/*.txt 2 | include mmtrack/.mim/model-index.yml 3 | recursive-include mmtrack/.mim/configs *.py *.yml 4 | recursive-include mmtrack/.mim/tools *.sh *.py 5 | -------------------------------------------------------------------------------- /software/mmtracking/configs/mot/bytetrack/bytetrack_yolox_x_crowdhuman_mot17-private.py: -------------------------------------------------------------------------------- 1 | _base_ = ['./bytetrack_yolox_x_crowdhuman_mot17-private-half.py'] 2 | 3 | data = dict( 4 | test=dict( 5 | ann_file='data/MOT17/annotations/test_cocoformat.json', 6 | img_prefix='data/MOT17/test')) 7 | -------------------------------------------------------------------------------- /software/mmtracking/configs/sot/siamese_rpn/siamese_rpn_r50_fp16_20e_lasot.py: -------------------------------------------------------------------------------- 1 | _base_ = ['./siamese_rpn_r50_20e_lasot.py'] 2 | optimizer_config = dict(type='SiameseRPNFp16OptimizerHook') 3 | fp16 = dict(loss_scale=512.) 4 | -------------------------------------------------------------------------------- /software/mmtracking/configs/vid/selsa/selsa_faster_rcnn_r50_dc5_fp16_1x_imagenetvid.py: -------------------------------------------------------------------------------- 1 | _base_ = ['./selsa_faster_rcnn_r50_dc5_1x_imagenetvid.py'] 2 | fp16 = dict(loss_scale=512.) 3 | -------------------------------------------------------------------------------- /software/mmtracking/demo/README.md: -------------------------------------------------------------------------------- 1 | # Demo 2 | 3 | Please refer to the inference of [VID](../docs/en/quick_run.md#inference-vid-models), [MOT](../docs/en/quick_run.md#inference-motvis-models), [SOT](../docs/en/quick_run.md#inference-sot-models) and [VIS](../docs/en/quick_run.md#inference-motvis-models) to run demos. 4 | -------------------------------------------------------------------------------- /software/mmtracking/demo/demo.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenRobotLab/DORT/5c1f39259247813a5c340753dd1a274f253a39b7/software/mmtracking/demo/demo.mp4 -------------------------------------------------------------------------------- /software/mmtracking/docs/en/_static/css/readthedocs.css: -------------------------------------------------------------------------------- 1 | .header-logo { 2 | background-image: url("../image/mmtrack-logo.png"); 3 | background-size: 145px 40px; 4 | height: 40px; 5 | width: 145px; 6 | } 7 | -------------------------------------------------------------------------------- /software/mmtracking/docs/en/switch_language.md: -------------------------------------------------------------------------------- 1 | ## English 2 | 3 | ## 简体中文 4 | -------------------------------------------------------------------------------- /software/mmtracking/docs/en/tutorials/index.rst: -------------------------------------------------------------------------------- 1 | .. toctree:: 2 | :maxdepth: 2 3 | 4 | config.md 5 | customize_dataset.md 6 | customize_data_pipeline.md 7 | customize_vid_model.md 8 | customize_mot_model.md 9 | customize_sot_model.md 10 | customize_runtime.md 11 | -------------------------------------------------------------------------------- /software/mmtracking/docs/zh_cn/_static/css/readthedocs.css: -------------------------------------------------------------------------------- 1 | .header-logo { 2 | background-image: url("../image/mmtrack-logo.png"); 3 | background-size: 145px 40px; 4 | height: 40px; 5 | width: 145px; 6 | } 7 | -------------------------------------------------------------------------------- /software/mmtracking/docs/zh_cn/switch_language.md: -------------------------------------------------------------------------------- 1 | ## English 2 | 3 | ## 简体中文 4 | -------------------------------------------------------------------------------- /software/mmtracking/docs/zh_cn/tutorials/index.rst: -------------------------------------------------------------------------------- 1 | .. toctree:: 2 | :maxdepth: 2 3 | 4 | config.md 5 | customize_dataset.md 6 | customize_data_pipeline.md 7 | customize_vid_model.md 8 | customize_mot_model.md 9 | customize_sot_model.md 10 | customize_runtime.md 11 | -------------------------------------------------------------------------------- /software/mmtracking/mmtrack/.mim/configs: -------------------------------------------------------------------------------- 1 | ../../configs -------------------------------------------------------------------------------- /software/mmtracking/mmtrack/.mim/model-index.yml: -------------------------------------------------------------------------------- 1 | ../../model-index.yml -------------------------------------------------------------------------------- /software/mmtracking/mmtrack/.mim/tools: -------------------------------------------------------------------------------- 1 | ../../tools -------------------------------------------------------------------------------- /software/mmtracking/mmtrack/core/anchor/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) OpenMMLab. All rights reserved. 2 | from .sot_anchor_generator import SiameseRPNAnchorGenerator 3 | 4 | __all__ = ['SiameseRPNAnchorGenerator'] 5 | -------------------------------------------------------------------------------- /software/mmtracking/mmtrack/core/bbox/iou_calculators/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) OpenMMLab. All rights reserved. 2 | from .region_iou_calculator import calculate_region_overlap 3 | 4 | __all__ = ['calculate_region_overlap'] 5 | -------------------------------------------------------------------------------- /software/mmtracking/mmtrack/core/hook/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) OpenMMLab. All rights reserved. 2 | from .yolox_mode_switch_hook import YOLOXModeSwitchHook 3 | 4 | __all__ = ['YOLOXModeSwitchHook'] 5 | -------------------------------------------------------------------------------- /software/mmtracking/mmtrack/core/motion/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) OpenMMLab. All rights reserved. 2 | from .flow import flow_warp_feats 3 | 4 | __all__ = ['flow_warp_feats'] 5 | -------------------------------------------------------------------------------- /software/mmtracking/mmtrack/datasets/parsers/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) OpenMMLab. All rights reserved. 2 | from .coco_video_parser import CocoVID 3 | 4 | __all__ = ['CocoVID'] 5 | -------------------------------------------------------------------------------- /software/mmtracking/mmtrack/models/aggregators/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) OpenMMLab. All rights reserved. 2 | from .embed_aggregator import EmbedAggregator 3 | from .selsa_aggregator import SelsaAggregator 4 | 5 | __all__ = ['EmbedAggregator', 'SelsaAggregator'] 6 | -------------------------------------------------------------------------------- /software/mmtracking/mmtrack/models/backbones/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) OpenMMLab. All rights reserved. 2 | from .sot_resnet import SOTResNet 3 | 4 | __all__ = ['SOTResNet'] 5 | -------------------------------------------------------------------------------- /software/mmtracking/mmtrack/models/losses/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) OpenMMLab. All rights reserved. 2 | from .l2_loss import L2Loss 3 | from .multipos_cross_entropy_loss import MultiPosCrossEntropyLoss 4 | from .triplet_loss import TripletLoss 5 | 6 | __all__ = ['L2Loss', 'TripletLoss', 'MultiPosCrossEntropyLoss'] 7 | -------------------------------------------------------------------------------- /software/mmtracking/mmtrack/models/roi_heads/bbox_heads/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) OpenMMLab. All rights reserved. 2 | from .selsa_bbox_head import SelsaBBoxHead 3 | 4 | __all__ = ['SelsaBBoxHead'] 5 | -------------------------------------------------------------------------------- /software/mmtracking/mmtrack/models/roi_heads/roi_extractors/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) OpenMMLab. All rights reserved. 2 | from .single_level_roi_extractor import SingleRoIExtractor 3 | from .temporal_roi_align import TemporalRoIAlign 4 | 5 | __all__ = ['SingleRoIExtractor', 'TemporalRoIAlign'] 6 | -------------------------------------------------------------------------------- /software/mmtracking/mmtrack/models/sot/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) OpenMMLab. All rights reserved. 2 | from .siamrpn import SiamRPN 3 | from .stark import Stark 4 | 5 | __all__ = ['SiamRPN', 'Stark'] 6 | -------------------------------------------------------------------------------- /software/mmtracking/mmtrack/models/vid/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) OpenMMLab. All rights reserved. 2 | from .base import BaseVideoDetector 3 | from .dff import DFF 4 | from .fgfa import FGFA 5 | from .selsa import SELSA 6 | 7 | __all__ = ['BaseVideoDetector', 'DFF', 'FGFA', 'SELSA'] 8 | -------------------------------------------------------------------------------- /software/mmtracking/mmtrack/utils/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) OpenMMLab. All rights reserved. 2 | from .collect_env import collect_env 3 | from .logger import get_root_logger 4 | 5 | __all__ = ['collect_env', 'get_root_logger'] 6 | -------------------------------------------------------------------------------- /software/mmtracking/requirements.txt: -------------------------------------------------------------------------------- 1 | -r requirements/build.txt 2 | -r requirements/runtime.txt 3 | -r requirements/tests.txt 4 | -------------------------------------------------------------------------------- /software/mmtracking/requirements/build.txt: -------------------------------------------------------------------------------- 1 | cython 2 | numba==0.53.0 3 | numpy 4 | -------------------------------------------------------------------------------- /software/mmtracking/requirements/docs.txt: -------------------------------------------------------------------------------- 1 | myst_parser 2 | -e git+https://github.com/open-mmlab/pytorch_sphinx_theme.git#egg=pytorch_sphinx_theme 3 | sphinx==4.0.2 4 | sphinx-copybutton 5 | sphinx_markdown_tables 6 | -------------------------------------------------------------------------------- /software/mmtracking/requirements/mminstall.txt: -------------------------------------------------------------------------------- 1 | mmcls>=0.16.0 2 | mmcv-full>=1.3.17,<1.6.0 3 | mmdet>=2.19.1,<3.0.0 4 | -------------------------------------------------------------------------------- /software/mmtracking/requirements/readthedocs.txt: -------------------------------------------------------------------------------- 1 | mmcls 2 | mmcv 3 | mmdet 4 | torch 5 | torchvision 6 | -------------------------------------------------------------------------------- /software/mmtracking/requirements/runtime.txt: -------------------------------------------------------------------------------- 1 | attributee==0.1.5 2 | dotty_dict 3 | lap 4 | matplotlib 5 | mmcls>=0.16.0 6 | motmetrics 7 | packaging 8 | pandas<=1.3.5 9 | pycocotools<=2.0.2 10 | seaborn 11 | terminaltables 12 | tqdm 13 | -------------------------------------------------------------------------------- /software/mmtracking/requirements/tests.txt: -------------------------------------------------------------------------------- 1 | asynctest 2 | codecov 3 | flake8 4 | interrogate 5 | isort==4.3.21 6 | # Note: used for kwarray.group_items, this may be ported to mmcv in the future. 7 | kwarray 8 | pytest 9 | ubelt 10 | xdoctest>=0.10.0 11 | yapf 12 | -------------------------------------------------------------------------------- /software/mmtracking/tests/test_data/test_datasets/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) OpenMMLab. All rights reserved. 2 | from .utils import _create_coco_gt_results 3 | 4 | __all__ = ['_create_coco_gt_results'] 5 | -------------------------------------------------------------------------------- /software/mmtracking/tests/test_models/test_forward/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) OpenMMLab. All rights reserved. 2 | from .utils import _demo_mm_inputs, _get_config_module 3 | 4 | __all__ = ['_demo_mm_inputs', '_get_config_module'] 5 | -------------------------------------------------------------------------------- /tools/data_converter/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) OpenMMLab. All rights reserved. 2 | -------------------------------------------------------------------------------- /tools/debug.sh: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env bash 2 | 3 | PORT=`expr $RANDOM % 40000 + 1000` 4 | echo $PORT 5 | 6 | -------------------------------------------------------------------------------- /tools/dist_eval_loss.sh: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env bash 2 | 3 | CONFIG=$1 4 | CHECKPOINT=$2 5 | 6 | PYTHONPATH="$(dirname $0)/..":$PYTHONPATH python tools/test.py $1 $2 --eval loss ${@:5} 7 | -------------------------------------------------------------------------------- /tools/test.sh: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env bash 2 | CONFIG=$1 3 | CHECKPOINT=$2 4 | 5 | PYTHONPATH="$(dirname $0)/..":$PYTHONPATH \ 6 | python3 $(dirname "$0")/test.py $CONFIG $CHECKPOINT --launcher pytorch ${@:3} 7 | --------------------------------------------------------------------------------