├── det3d ├── ops │ ├── __init__.py │ └── bev_pool_v2 │ │ └── __init__.py ├── models │ ├── backbones │ │ ├── DCNv2 │ │ │ ├── __init__.py │ │ │ ├── make.sh │ │ │ └── .gitignore │ │ └── DCNv2_t18 │ │ │ ├── __init__.py │ │ │ └── make.sh │ ├── losses │ │ └── __init__.py │ ├── motion │ │ └── __init__.py │ ├── track_heads │ │ └── __init__.py │ ├── track_frameworks │ │ └── __init__.py │ └── trackers │ │ └── __init__.py ├── version.py ├── apis │ └── __init__.py ├── core │ ├── bbox │ │ ├── assigners │ │ │ └── __init__.py │ │ ├── coders │ │ │ └── __init__.py │ │ └── match_costs │ │ │ └── __init__.py │ └── visualizer │ │ └── __init__.py └── hooks │ └── __init__.py ├── software ├── mmdet │ ├── mmdet │ │ ├── .mim │ │ │ ├── demo │ │ │ ├── tools │ │ │ ├── configs │ │ │ └── model-index.yml │ │ ├── models │ │ │ ├── roi_heads │ │ │ │ └── shared_heads │ │ │ │ │ └── __init__.py │ │ │ └── seg_heads │ │ │ │ └── __init__.py │ │ └── core │ │ │ ├── data_structures │ │ │ └── __init__.py │ │ │ └── bbox │ │ │ └── iou_calculators │ │ │ └── __init__.py │ ├── docs │ │ ├── zh_cn │ │ │ ├── useful_tools.md │ │ │ ├── tutorials │ │ │ │ ├── customize_runtime.md │ │ │ │ └── pytorch2onnx.md │ │ │ ├── switch_language.md │ │ │ └── _static │ │ │ │ └── css │ │ │ │ └── readthedocs.css │ │ └── en │ │ │ ├── switch_language.md │ │ │ └── _static │ │ │ └── css │ │ │ └── readthedocs.css │ ├── requirements │ │ ├── mminstall.txt │ │ ├── readthedocs.txt │ │ ├── albu.txt │ │ ├── runtime.txt │ │ ├── build.txt │ │ ├── optional.txt │ │ └── docs.txt │ ├── demo │ │ └── demo.mp4 │ ├── .dev_scripts │ │ └── linter.sh │ ├── requirements.txt │ ├── configs │ │ ├── centernet │ │ │ └── centernet_resnet18_140e_coco.py │ │ ├── mask_rcnn │ │ │ ├── mask_rcnn_r50_fpn_fp16_1x_coco.py │ │ │ ├── mask_rcnn_r50_fpn_mstrain-poly_3x_coco.py │ │ │ ├── mask_rcnn_r50_fpn_1x_coco.py │ │ │ ├── mask_rcnn_r50_fpn_2x_coco.py │ │ │ ├── mask_rcnn_r50_caffe_fpn_mstrain-poly_2x_coco.py │ │ │ ├── mask_rcnn_r50_caffe_fpn_mstrain-poly_3x_coco.py │ │ │ ├── mask_rcnn_r101_fpn_1x_coco.py │ │ │ ├── mask_rcnn_r101_fpn_2x_coco.py │ │ │ └── mask_rcnn_r101_caffe_fpn_1x_coco.py │ │ ├── retinanet │ │ │ ├── retinanet_r50_fpn_fp16_1x_coco.py │ │ │ ├── retinanet_r50_fpn_2x_coco.py │ │ │ ├── retinanet_r50_caffe_fpn_mstrain_2x_coco.py │ │ │ ├── retinanet_r50_caffe_fpn_mstrain_3x_coco.py │ │ │ ├── retinanet_r50_fpn_mstrain_640-800_3x_coco.py │ │ │ ├── retinanet_r101_fpn_1x_coco.py │ │ │ ├── retinanet_r101_fpn_2x_coco.py │ │ │ └── retinanet_r101_caffe_fpn_1x_coco.py │ │ ├── tood │ │ │ ├── tood_r50_fpn_anchor_based_1x_coco.py │ │ │ └── tood_r101_fpn_mstrain_2x_coco.py │ │ ├── faster_rcnn │ │ │ ├── faster_rcnn_r50_fpn_fp16_1x_coco.py │ │ │ ├── faster_rcnn_r50_fpn_mstrain_3x_coco.py │ │ │ ├── faster_rcnn_r50_fpn_ohem_1x_coco.py │ │ │ ├── faster_rcnn_r50_caffe_dc5_mstrain_3x_coco.py │ │ │ ├── faster_rcnn_r50_caffe_fpn_mstrain_2x_coco.py │ │ │ ├── faster_rcnn_r50_fpn_1x_coco.py │ │ │ ├── faster_rcnn_r50_fpn_2x_coco.py │ │ │ ├── faster_rcnn_r101_fpn_1x_coco.py │ │ │ ├── faster_rcnn_r101_fpn_2x_coco.py │ │ │ ├── faster_rcnn_r50_fpn_iou_1x_coco.py │ │ │ ├── faster_rcnn_r50_fpn_ciou_1x_coco.py │ │ │ ├── faster_rcnn_r50_fpn_giou_1x_coco.py │ │ │ ├── faster_rcnn_r101_fpn_mstrain_3x_coco.py │ │ │ ├── faster_rcnn_r50_fpn_bounded_iou_1x_coco.py │ │ │ └── faster_rcnn_r101_caffe_fpn_1x_coco.py │ │ ├── paa │ │ │ ├── paa_r50_fpn_2x_coco.py │ │ │ ├── paa_r101_fpn_2x_coco.py │ │ │ ├── paa_r50_fpn_1.5x_coco.py │ │ │ ├── paa_r101_fpn_1x_coco.py │ │ │ └── paa_r101_fpn_mstrain_3x_coco.py │ │ ├── yolo │ │ │ └── yolov3_d53_fp16_mstrain-608_273e_coco.py │ │ ├── deformable_detr │ │ │ ├── deformable_detr_refine_r50_16x2_50e_coco.py │ │ │ └── deformable_detr_twostage_refine_r50_16x2_50e_coco.py │ │ ├── cascade_rcnn │ │ │ ├── cascade_mask_rcnn_r50_fpn_mstrain_3x_coco.py │ │ │ ├── cascade_rcnn_r50_fpn_20e_coco.py │ │ │ ├── cascade_rcnn_r50_fpn_1x_coco.py │ │ │ ├── cascade_mask_rcnn_r50_fpn_1x_coco.py │ │ │ ├── cascade_mask_rcnn_r50_fpn_20e_coco.py │ │ │ ├── cascade_rcnn_r101_fpn_1x_coco.py │ │ │ ├── cascade_rcnn_r101_fpn_20e_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_rcnn_r101_caffe_fpn_1x_coco.py │ │ ├── fcos │ │ │ ├── fcos_center_r50_caffe_fpn_gn-head_1x_coco.py │ │ │ ├── fcos_r50_caffe_fpn_gn-head_4x4_1x_coco.py │ │ │ └── fcos_r101_caffe_fpn_gn-head_1x_coco.py │ │ ├── htc │ │ │ └── htc_r50_fpn_20e_coco.py │ │ ├── reppoints │ │ │ ├── reppoints_minmax_r50_fpn_gn-neck+head_1x_coco.py │ │ │ ├── reppoints_partial_minmax_r50_fpn_gn-neck+head_1x_coco.py │ │ │ ├── bbox_r50_grid_center_fpn_gn-neck+head_1x_coco.py │ │ │ ├── reppoints_moment_r50_fpn_gn-neck+head_2x_coco.py │ │ │ ├── reppoints_moment_r50_fpn_gn-neck+head_1x_coco.py │ │ │ └── reppoints_moment_r101_fpn_gn-neck+head_2x_coco.py │ │ ├── rpn │ │ │ ├── rpn_r50_fpn_2x_coco.py │ │ │ ├── rpn_r101_fpn_1x_coco.py │ │ │ ├── rpn_r101_fpn_2x_coco.py │ │ │ └── rpn_r101_caffe_fpn_1x_coco.py │ │ ├── scnet │ │ │ ├── scnet_r50_fpn_20e_coco.py │ │ │ └── scnet_r101_fpn_20e_coco.py │ │ ├── strong_baselines │ │ │ ├── mask_rcnn_r50_caffe_fpn_syncbn-all_rpn-2conv_lsj_100e_fp16_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 │ │ ├── foveabox │ │ │ ├── fovea_r50_fpn_4x4_2x_coco.py │ │ │ ├── fovea_r101_fpn_4x4_1x_coco.py │ │ │ └── fovea_r101_fpn_4x4_2x_coco.py │ │ ├── hrnet │ │ │ ├── htc_hrnetv2p_w40_28e_coco.py │ │ │ ├── faster_rcnn_hrnetv2p_w32_2x_coco.py │ │ │ ├── faster_rcnn_hrnetv2p_w40_2x_coco.py │ │ │ ├── mask_rcnn_hrnetv2p_w18_2x_coco.py │ │ │ ├── mask_rcnn_hrnetv2p_w32_2x_coco.py │ │ │ ├── mask_rcnn_hrnetv2p_w40_2x_coco.py │ │ │ ├── faster_rcnn_hrnetv2p_w18_2x_coco.py │ │ │ ├── htc_x101_64x4d_fpn_16x1_28e_coco.py │ │ │ ├── fcos_hrnetv2p_w18_gn-head_4x4_2x_coco.py │ │ │ └── fcos_hrnetv2p_w32_gn-head_4x4_2x_coco.py │ │ ├── regnet │ │ │ └── faster_rcnn_regnetx-3.2GF_fpn_2x_coco.py │ │ ├── fast_rcnn │ │ │ ├── fast_rcnn_r50_fpn_2x_coco.py │ │ │ ├── fast_rcnn_r101_fpn_1x_coco.py │ │ │ ├── fast_rcnn_r101_fpn_2x_coco.py │ │ │ └── fast_rcnn_r101_caffe_fpn_1x_coco.py │ │ ├── tridentnet │ │ │ └── tridentnet_r50_caffe_mstrain_3x_coco.py │ │ ├── ms_rcnn │ │ │ ├── ms_rcnn_r101_caffe_fpn_2x_coco.py │ │ │ ├── ms_rcnn_r50_caffe_fpn_2x_coco.py │ │ │ ├── ms_rcnn_x101_64x4d_fpn_2x_coco.py │ │ │ └── ms_rcnn_r101_caffe_fpn_1x_coco.py │ │ ├── gn │ │ │ ├── mask_rcnn_r101_fpn_gn-all_3x_coco.py │ │ │ ├── mask_rcnn_r50_fpn_gn-all_3x_coco.py │ │ │ ├── mask_rcnn_r50_fpn_gn-all_contrib_3x_coco.py │ │ │ └── mask_rcnn_r101_fpn_gn-all_2x_coco.py │ │ ├── solov2 │ │ │ ├── solov2_r101_fpn_3x_coco.py │ │ │ ├── solov2_light_r18_fpn_3x_coco.py │ │ │ └── solov2_light_r34_fpn_3x_coco.py │ │ ├── fpg │ │ │ └── retinanet_r50_fpg-chn128_crop640_50e_coco.py │ │ ├── gn+ws │ │ │ ├── mask_rcnn_r101_fpn_gn_ws-all_20_23_24e_coco.py │ │ │ ├── mask_rcnn_r50_fpn_gn_ws-all_20_23_24e_coco.py │ │ │ ├── mask_rcnn_x50_32x4d_fpn_gn_ws-all_20_23_24e_coco.py │ │ │ ├── mask_rcnn_x101_32x4d_fpn_gn_ws-all_20_23_24e_coco.py │ │ │ ├── faster_rcnn_r101_fpn_gn_ws-all_1x_coco.py │ │ │ └── mask_rcnn_r101_fpn_gn_ws-all_2x_coco.py │ │ ├── gcnet │ │ │ ├── mask_rcnn_r101_fpn_syncbn-backbone_1x_coco.py │ │ │ ├── mask_rcnn_r50_fpn_syncbn-backbone_1x_coco.py │ │ │ ├── mask_rcnn_x101_32x4d_fpn_syncbn-backbone_1x_coco.py │ │ │ ├── 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 │ │ ├── point_rend │ │ │ └── point_rend_r50_caffe_fpn_mstrain_3x_coco.py │ │ ├── swin │ │ │ └── mask_rcnn_swin-t-p4-w7_fpn_fp16_ms-crop-3x_coco.py │ │ ├── atss │ │ │ └── atss_r101_fpn_1x_coco.py │ │ ├── fsaf │ │ │ └── fsaf_r101_fpn_1x_coco.py │ │ ├── openimages │ │ │ ├── faster_rcnn_r50_fpn_32x2_cas_1x_openimages.py │ │ │ └── faster_rcnn_r50_fpn_32x2_cas_1x_openimages_challenge.py │ │ ├── vfnet │ │ │ ├── vfnet_r101_fpn_1x_coco.py │ │ │ └── vfnet_r101_fpn_mstrain_2x_coco.py │ │ ├── yolact │ │ │ └── yolact_r101_1x8_coco.py │ │ ├── ghm │ │ │ └── retinanet_ghm_r101_fpn_1x_coco.py │ │ ├── pafpn │ │ │ └── faster_rcnn_r50_pafpn_1x_coco.py │ │ ├── panoptic_fpn │ │ │ ├── panoptic_fpn_r101_fpn_1x_coco.py │ │ │ └── panoptic_fpn_r101_fpn_mstrain_3x_coco.py │ │ ├── dcn │ │ │ ├── mask_rcnn_r50_fpn_dconv_c3-c5_1x_coco.py │ │ │ ├── faster_rcnn_r50_fpn_dconv_c3-c5_1x_coco.py │ │ │ ├── mask_rcnn_r101_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 │ │ │ ├── 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 │ │ ├── dcnv2 │ │ │ ├── mask_rcnn_r50_fpn_mdconv_c3-c5_1x_coco.py │ │ │ ├── faster_rcnn_r50_fpn_mdconv_c3-c5_1x_coco.py │ │ │ └── faster_rcnn_r50_fpn_mdconv_c3-c5_group4_1x_coco.py │ │ ├── libra_rcnn │ │ │ └── libra_faster_rcnn_r101_fpn_1x_coco.py │ │ ├── grid_rcnn │ │ │ └── grid_rcnn_r101_fpn_gn-head_2x_coco.py │ │ ├── instaboost │ │ │ ├── mask_rcnn_r101_fpn_instaboost_4x_coco.py │ │ │ └── cascade_mask_rcnn_r101_fpn_instaboost_4x_coco.py │ │ ├── mask2former │ │ │ ├── mask2former_r101_lsj_8x2_50e_coco.py │ │ │ └── mask2former_r101_lsj_8x2_50e_coco-panoptic.py │ │ ├── free_anchor │ │ │ └── retinanet_free_anchor_r101_fpn_1x_coco.py │ │ ├── pvt │ │ │ ├── retinanet_pvt-s_fpn_1x_coco.py │ │ │ └── retinanet_pvt-m_fpn_1x_coco.py │ │ ├── lvis │ │ │ ├── mask_rcnn_r101_fpn_sample1e-3_mstrain_1x_lvis_v1.py │ │ │ └── mask_rcnn_r101_fpn_sample1e-3_mstrain_2x_lvis_v0.5.py │ │ ├── queryinst │ │ │ └── queryinst_r101_fpn_mstrain_480-800_3x_coco.py │ │ ├── guided_anchoring │ │ │ ├── ga_faster_r101_caffe_fpn_1x_coco.py │ │ │ └── ga_retinanet_r101_caffe_fpn_1x_coco.py │ │ ├── pisa │ │ │ ├── pisa_ssd300_coco.py │ │ │ └── pisa_ssd512_coco.py │ │ ├── seesaw_loss │ │ │ ├── mask_rcnn_r50_fpn_random_seesaw_loss_normed_mask_mstrain_2x_lvis_v1.py │ │ │ └── mask_rcnn_r50_fpn_sample1e-3_seesaw_loss_normed_mask_mstrain_2x_lvis_v1.py │ │ ├── sparse_rcnn │ │ │ └── sparse_rcnn_r101_fpn_mstrain_480-800_3x_coco.py │ │ └── detectors │ │ │ └── htc_r50_sac_1x_coco.py │ ├── tests │ │ ├── test_onnx │ │ │ └── __init__.py │ │ ├── test_models │ │ │ ├── test_roi_heads │ │ │ │ └── __init__.py │ │ │ └── test_backbones │ │ │ │ └── __init__.py │ │ └── test_data │ │ │ └── test_pipelines │ │ │ └── test_transform │ │ │ └── __init__.py │ ├── .readthedocs.yml │ ├── docker │ │ └── serve │ │ │ ├── config.properties │ │ │ └── entrypoint.sh │ ├── MANIFEST.in │ ├── .owners.yml │ └── CITATION.cff ├── mmdet3d │ ├── requirements │ │ ├── build.txt │ │ ├── optional.txt │ │ ├── mminstall.txt │ │ ├── readthedocs.txt │ │ ├── docs.txt │ │ ├── runtime.txt │ │ └── tests.txt │ ├── docs │ │ ├── zh_cn │ │ │ ├── changelog.md │ │ │ ├── compatibility.md │ │ │ ├── supported_tasks │ │ │ │ └── index.rst │ │ │ ├── switch_language.md │ │ │ ├── _static │ │ │ │ └── css │ │ │ │ │ └── readthedocs.css │ │ │ ├── datasets │ │ │ │ └── index.rst │ │ │ └── tutorials │ │ │ │ └── index.rst │ │ └── en │ │ │ ├── supported_tasks │ │ │ └── index.rst │ │ │ ├── switch_language.md │ │ │ ├── _static │ │ │ └── css │ │ │ │ └── readthedocs.css │ │ │ └── datasets │ │ │ └── index.rst │ ├── .github │ │ ├── ISSUE_TEMPLATE │ │ │ ├── config.yml │ │ │ └── general_questions.md │ │ └── CONTRIBUTING.md │ ├── tools │ │ └── data_converter │ │ │ └── __init__.py │ ├── .dev_scripts │ │ └── linter.sh │ ├── resources │ │ ├── nuimages_demo.gif │ │ ├── open3d_visual.gif │ │ └── mmdet3d_outdoor_demo.gif │ ├── requirements.txt │ ├── mmdet3d │ │ ├── core │ │ │ ├── evaluation │ │ │ │ ├── waymo_utils │ │ │ │ │ ├── compute_detection_let_metrics_main │ │ │ │ │ └── __init__.py │ │ │ │ ├── kitti_utils │ │ │ │ │ └── __init__.py │ │ │ │ └── scannet_utils │ │ │ │ │ └── __init__.py │ │ │ ├── bbox │ │ │ │ └── assigners │ │ │ │ │ └── __init__.py │ │ │ ├── voxel │ │ │ │ └── __init__.py │ │ │ └── visualizer │ │ │ │ └── __init__.py │ │ ├── ops │ │ │ ├── paconv │ │ │ │ └── __init__.py │ │ │ ├── spconv │ │ │ │ └── overwrite_spconv │ │ │ │ │ └── __init__.py │ │ │ └── dgcnn_modules │ │ │ │ └── __init__.py │ │ └── models │ │ │ ├── segmentors │ │ │ └── __init__.py │ │ │ ├── roi_heads │ │ │ └── mask_heads │ │ │ │ └── __init__.py │ │ │ ├── decode_heads │ │ │ └── __init__.py │ │ │ └── model_utils │ │ │ └── __init__.py │ ├── tests │ │ └── test_samples │ │ │ └── parta2_roihead_inputs.npz │ ├── configs │ │ ├── second │ │ │ ├── hv_second_secfpn_fp16_6x8_80e_kitti-3d-car.py │ │ │ ├── hv_second_secfpn_fp16_6x8_80e_kitti-3d-3class.py │ │ │ └── hv_second_secfpn_6x8_80e_kitti-3d-3class.py │ │ ├── nuimages │ │ │ ├── mask_rcnn_r101_fpn_1x_nuim.py │ │ │ ├── cascade_mask_rcnn_r101_fpn_1x_nuim.py │ │ │ ├── htc_r50_fpn_coco-20e_20e_nuim.py │ │ │ └── htc_r50_fpn_coco-20e_1x_nuim.py │ │ ├── centerpoint │ │ │ ├── centerpoint_01voxel_second_secfpn_circlenms_4x8_cyclic_20e_nus.py │ │ │ ├── centerpoint_0075voxel_second_secfpn_circlenms_4x8_cyclic_20e_nus.py │ │ │ └── centerpoint_02pillar_second_secfpn_circlenms_4x8_cyclic_20e_nus.py │ │ ├── _base_ │ │ │ └── models │ │ │ │ └── paconv_cuda_ssg.py │ │ ├── pgd │ │ │ └── pgd_r101_caffe_fpn_gn-head_2x16_2x_nus-mono3d.py │ │ ├── pointpillars │ │ │ ├── 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_sbn-all_fp16_2x8_2x_nus-3d.py │ │ │ └── hv_pointpillars_secfpn_sbn_2x16_2x_waymoD5-3d-3class.py │ │ └── regnet │ │ │ └── hv_pointpillars_regnet-400mf_fpn_sbn-all_fp16_2x8_2x_nus-3d.py │ ├── docker │ │ └── serve │ │ │ ├── config.properties │ │ │ └── entrypoint.sh │ ├── MANIFEST.in │ └── .readthedocs.yml ├── mmseg │ ├── mmseg │ │ ├── .mim │ │ │ ├── tools │ │ │ ├── configs │ │ │ └── model-index.yml │ │ ├── ops │ │ │ └── __init__.py │ │ └── core │ │ │ ├── __init__.py │ │ │ ├── seg │ │ │ ├── sampler │ │ │ │ └── __init__.py │ │ │ └── __init__.py │ │ │ └── utils │ │ │ └── __init__.py │ ├── requirements │ │ ├── optional.txt │ │ ├── mminstall.txt │ │ ├── readthedocs.txt │ │ ├── runtime.txt │ │ ├── tests.txt │ │ └── docs.txt │ ├── tests │ │ ├── __init__.py │ │ └── test_models │ │ │ ├── __init__.py │ │ │ ├── test_heads │ │ │ └── __init__.py │ │ │ ├── test_losses │ │ │ └── __init__.py │ │ │ ├── test_necks │ │ │ └── __init__.py │ │ │ ├── test_utils │ │ │ └── __init__.py │ │ │ ├── test_segmentors │ │ │ └── __init__.py │ │ │ └── test_backbones │ │ │ └── __init__.py │ ├── requirements.txt │ ├── resources │ │ └── seg_demo.gif │ ├── configs │ │ ├── stdc │ │ │ └── stdc2_512x1024_80k_cityscapes.py │ │ ├── icnet │ │ │ ├── icnet_r101-d8_832x832_160k_cityscapes.py │ │ │ ├── icnet_r101-d8_832x832_80k_cityscapes.py │ │ │ ├── icnet_r18-d8_832x832_160k_cityscapes.py │ │ │ ├── icnet_r18-d8_832x832_80k_cityscapes.py │ │ │ ├── icnet_r50-d8_832x832_80k_cityscapes.py │ │ │ ├── icnet_r50-d8_832x832_160k_cityscapes.py │ │ │ ├── icnet_r50-d8_in1k-pre_832x832_80k_cityscapes.py │ │ │ └── icnet_r50-d8_in1k-pre_832x832_160k_cityscapes.py │ │ ├── setr │ │ │ └── setr_mla_512x512_160k_b16_ade20k.py │ │ ├── ann │ │ │ ├── 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_r101-d8_512x1024_40k_cityscapes.py │ │ │ ├── ann_r101-d8_512x1024_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_80k_ade20k.py │ │ ├── dnlnet │ │ │ ├── dnl_r101-d8_512x512_80k_ade20k.py │ │ │ ├── dnl_r101-d8_512x512_160k_ade20k.py │ │ │ ├── dnl_r101-d8_512x1024_40k_cityscapes.py │ │ │ ├── dnl_r101-d8_512x1024_80k_cityscapes.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 │ │ ├── fcn │ │ │ ├── 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_r50b-d8_769x769_80k_cityscapes.py │ │ │ ├── fcn_r101-d8_512x1024_40k_cityscapes.py │ │ │ ├── fcn_r101-d8_512x1024_80k_cityscapes.py │ │ │ ├── fcn_r50b-d8_512x1024_80k_cityscapes.py │ │ │ ├── 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_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_80k_pascal_context.py │ │ │ ├── fcn_r101-d8_480x480_40k_pascal_context_59.py │ │ │ ├── fcn_r101-d8_480x480_80k_pascal_context_59.py │ │ │ ├── fcn_r101b-d8_769x769_80k_cityscapes.py │ │ │ ├── fcn_r101b-d8_512x1024_80k_cityscapes.py │ │ │ ├── fcn_d6_r101b-d16_512x1024_80k_cityscapes.py │ │ │ ├── fcn_d6_r101b-d16_769x769_80k_cityscapes.py │ │ │ ├── fcn_r50-d8_512x1024_40k_cityscapes.py │ │ │ ├── fcn_r50-d8_512x1024_80k_cityscapes.py │ │ │ └── fcn_r101-d8_fp16_512x1024_80k_cityscapes.py │ │ ├── sem_fpn │ │ │ ├── fpn_r101_512x512_160k_ade20k.py │ │ │ ├── fpn_r101_512x1024_80k_cityscapes.py │ │ │ ├── fpn_r50_512x1024_80k_cityscapes.py │ │ │ └── fpn_r50_512x512_160k_ade20k.py │ │ ├── apcnet │ │ │ ├── apcnet_r101-d8_512x512_80k_ade20k.py │ │ │ ├── apcnet_r101-d8_512x512_160k_ade20k.py │ │ │ ├── apcnet_r101-d8_512x1024_40k_cityscapes.py │ │ │ ├── apcnet_r101-d8_512x1024_80k_cityscapes.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 │ │ ├── ccnet │ │ │ ├── ccnet_r101-d8_512x512_160k_ade20k.py │ │ │ ├── ccnet_r101-d8_512x512_80k_ade20k.py │ │ │ ├── ccnet_r101-d8_512x512_20k_voc12aug.py │ │ │ ├── ccnet_r101-d8_512x512_40k_voc12aug.py │ │ │ ├── ccnet_r101-d8_769x769_40k_cityscapes.py │ │ │ ├── ccnet_r101-d8_769x769_80k_cityscapes.py │ │ │ ├── ccnet_r101-d8_512x1024_40k_cityscapes.py │ │ │ ├── ccnet_r101-d8_512x1024_80k_cityscapes.py │ │ │ ├── ccnet_r50-d8_512x1024_40k_cityscapes.py │ │ │ └── ccnet_r50-d8_512x1024_80k_cityscapes.py │ │ ├── danet │ │ │ ├── danet_r101-d8_512x512_160k_ade20k.py │ │ │ ├── danet_r101-d8_512x512_80k_ade20k.py │ │ │ ├── danet_r101-d8_512x512_20k_voc12aug.py │ │ │ ├── danet_r101-d8_512x512_40k_voc12aug.py │ │ │ ├── danet_r101-d8_769x769_40k_cityscapes.py │ │ │ ├── danet_r101-d8_769x769_80k_cityscapes.py │ │ │ ├── danet_r101-d8_512x1024_40k_cityscapes.py │ │ │ ├── danet_r101-d8_512x1024_80k_cityscapes.py │ │ │ ├── danet_r50-d8_512x1024_40k_cityscapes.py │ │ │ └── danet_r50-d8_512x1024_80k_cityscapes.py │ │ ├── dmnet │ │ │ ├── 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_r101-d8_512x1024_40k_cityscapes.py │ │ │ ├── dmnet_r101-d8_512x1024_80k_cityscapes.py │ │ │ ├── dmnet_r50-d8_512x1024_40k_cityscapes.py │ │ │ └── dmnet_r50-d8_512x1024_80k_cityscapes.py │ │ ├── encnet │ │ │ ├── encnet_r101-d8_512x512_80k_ade20k.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_512x1024_40k_cityscapes.py │ │ │ ├── encnet_r101-d8_512x1024_80k_cityscapes.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 │ │ ├── gcnet │ │ │ ├── gcnet_r101-d8_512x512_160k_ade20k.py │ │ │ ├── gcnet_r101-d8_512x512_80k_ade20k.py │ │ │ ├── gcnet_r101-d8_512x512_20k_voc12aug.py │ │ │ ├── gcnet_r101-d8_512x512_40k_voc12aug.py │ │ │ ├── gcnet_r101-d8_769x769_40k_cityscapes.py │ │ │ ├── gcnet_r101-d8_769x769_80k_cityscapes.py │ │ │ ├── gcnet_r101-d8_512x1024_40k_cityscapes.py │ │ │ ├── gcnet_r101-d8_512x1024_80k_cityscapes.py │ │ │ ├── gcnet_r50-d8_512x1024_40k_cityscapes.py │ │ │ └── gcnet_r50-d8_512x1024_80k_cityscapes.py │ │ ├── isanet │ │ │ ├── isanet_r101-d8_512x512_80k_ade20k.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_512x1024_40k_cityscapes.py │ │ │ ├── isanet_r101-d8_512x1024_80k_cityscapes.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 │ │ ├── psanet │ │ │ ├── psanet_r101-d8_512x512_80k_ade20k.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_512x1024_40k_cityscapes.py │ │ │ ├── psanet_r101-d8_512x1024_80k_cityscapes.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 │ │ ├── pspnet │ │ │ ├── pspnet_r101-d8_512x1024_40k_dark.py │ │ │ ├── pspnet_r101-d8_512x512_80k_ade20k.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_4x4_512x512_80k_potsdam.py │ │ │ ├── pspnet_r101-d8_512x1024_40k_cityscapes.py │ │ │ ├── pspnet_r101-d8_512x1024_80k_cityscapes.py │ │ │ ├── pspnet_r101-d8_769x769_40k_cityscapes.py │ │ │ ├── pspnet_r101-d8_769x769_80k_cityscapes.py │ │ │ ├── pspnet_r50b-d8_512x1024_80k_cityscapes.py │ │ │ ├── pspnet_r50b-d8_769x769_80k_cityscapes.py │ │ │ ├── pspnet_r101-d8_4x4_512x512_80k_vaihingen.py │ │ │ ├── pspnet_r101-d8_480x480_40k_pascal_context.py │ │ │ ├── pspnet_r101-d8_480x480_80k_pascal_context.py │ │ │ ├── pspnet_r101-d8_512x1024_40k_night_driving.py │ │ │ ├── pspnet_r101-d8_480x480_40k_pascal_context_59.py │ │ │ ├── pspnet_r101-d8_480x480_80k_pascal_context_59.py │ │ │ ├── pspnet_r101-d8_512x512_4x4_20k_coco-stuff10k.py │ │ │ ├── pspnet_r101-d8_512x512_4x4_40k_coco-stuff10k.py │ │ │ ├── pspnet_r101-d8_512x512_4x4_80k_coco-stuff164k.py │ │ │ ├── pspnet_r101-d8_512x512_4x4_160k_coco-stuff164k.py │ │ │ ├── pspnet_r101-d8_512x512_4x4_320k_coco-stuff164k.py │ │ │ ├── pspnet_r101b-d8_512x1024_80k_dark.py │ │ │ ├── pspnet_r101b-d8_512x1024_80k_cityscapes.py │ │ │ ├── pspnet_r101b-d8_769x769_80k_cityscapes.py │ │ │ ├── pspnet_r101b-d8_512x1024_80k_night_driving.py │ │ │ ├── pspnet_r50-d8_512x1024_40k_cityscapes.py │ │ │ ├── pspnet_r50-d8_512x1024_80k_cityscapes.py │ │ │ ├── pspnet_r101-d8_fp16_512x1024_80k_cityscapes.py │ │ │ └── pspnet_r101-d8_512x512_80k_loveda.py │ │ ├── upernet │ │ │ ├── upernet_r101_512x512_160k_ade20k.py │ │ │ ├── upernet_r101_512x512_80k_ade20k.py │ │ │ ├── upernet_r101_512x512_20k_voc12aug.py │ │ │ ├── upernet_r101_512x512_40k_voc12aug.py │ │ │ ├── upernet_r101_769x769_40k_cityscapes.py │ │ │ ├── upernet_r101_769x769_80k_cityscapes.py │ │ │ ├── upernet_r101_512x1024_40k_cityscapes.py │ │ │ ├── upernet_r101_512x1024_80k_cityscapes.py │ │ │ ├── upernet_r50_512x1024_40k_cityscapes.py │ │ │ └── upernet_r50_512x1024_80k_cityscapes.py │ │ ├── deeplabv3 │ │ │ ├── deeplabv3_r101-d8_512x512_80k_ade20k.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_512x1024_40k_cityscapes.py │ │ │ ├── deeplabv3_r101-d8_512x1024_80k_cityscapes.py │ │ │ ├── deeplabv3_r101-d8_769x769_40k_cityscapes.py │ │ │ ├── deeplabv3_r101-d8_769x769_80k_cityscapes.py │ │ │ ├── deeplabv3_r50b-d8_512x1024_80k_cityscapes.py │ │ │ ├── deeplabv3_r50b-d8_769x769_80k_cityscapes.py │ │ │ ├── deeplabv3_r101-d8_480x480_40k_pascal_context.py │ │ │ ├── deeplabv3_r101-d8_480x480_80k_pascal_context.py │ │ │ ├── deeplabv3_r101-d8_480x480_40k_pascal_context_59.py │ │ │ ├── deeplabv3_r101-d8_480x480_80k_pascal_context_59.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_r101b-d8_512x1024_80k_cityscapes.py │ │ │ ├── deeplabv3_r101b-d8_769x769_80k_cityscapes.py │ │ │ ├── deeplabv3_r50-d8_512x1024_40k_cityscapes.py │ │ │ ├── deeplabv3_r50-d8_512x1024_80k_cityscapes.py │ │ │ └── deeplabv3_r101-d8_fp16_512x1024_80k_cityscapes.py │ │ ├── emanet │ │ │ ├── emanet_r101-d8_512x1024_80k_cityscapes.py │ │ │ ├── emanet_r101-d8_769x769_80k_cityscapes.py │ │ │ └── emanet_r50-d8_512x1024_80k_cityscapes.py │ │ ├── nonlocal_net │ │ │ ├── nonlocal_r101-d8_512x512_80k_ade20k.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_512x1024_40k_cityscapes.py │ │ │ ├── nonlocal_r101-d8_512x1024_80k_cityscapes.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 │ │ ├── point_rend │ │ │ ├── pointrend_r101_512x512_160k_ade20k.py │ │ │ ├── pointrend_r101_512x1024_80k_cityscapes.py │ │ │ └── pointrend_r50_512x1024_80k_cityscapes.py │ │ ├── deeplabv3plus │ │ │ ├── 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_potsdam.py │ │ │ ├── deeplabv3plus_r101-d8_769x769_40k_cityscapes.py │ │ │ ├── deeplabv3plus_r101-d8_769x769_80k_cityscapes.py │ │ │ ├── deeplabv3plus_r50b-d8_769x769_80k_cityscapes.py │ │ │ ├── deeplabv3plus_r101-d8_4x4_512x512_80k_vaihingen.py │ │ │ ├── deeplabv3plus_r101-d8_512x1024_40k_cityscapes.py │ │ │ ├── deeplabv3plus_r101-d8_512x1024_80k_cityscapes.py │ │ │ ├── deeplabv3plus_r50b-d8_512x1024_80k_cityscapes.py │ │ │ ├── deeplabv3plus_r101-d8_480x480_40k_pascal_context.py │ │ │ ├── deeplabv3plus_r101-d8_480x480_80k_pascal_context.py │ │ │ ├── deeplabv3plus_r101-d8_480x480_40k_pascal_context_59.py │ │ │ ├── deeplabv3plus_r101-d8_480x480_80k_pascal_context_59.py │ │ │ ├── deeplabv3plus_r101b-d8_769x769_80k_cityscapes.py │ │ │ ├── deeplabv3plus_r101b-d8_512x1024_80k_cityscapes.py │ │ │ ├── deeplabv3plus_r50-d8_512x1024_40k_cityscapes.py │ │ │ ├── deeplabv3plus_r50-d8_512x1024_80k_cityscapes.py │ │ │ ├── deeplabv3plus_r101-d8_fp16_512x1024_80k_cityscapes.py │ │ │ └── deeplabv3plus_r101-d8_512x512_80k_loveda.py │ │ ├── hrnet │ │ │ ├── fcn_hr18_512x1024_160k_cityscapes.py │ │ │ ├── fcn_hr18_512x1024_40k_cityscapes.py │ │ │ ├── fcn_hr18_512x1024_80k_cityscapes.py │ │ │ ├── fcn_hr18_4x4_896x896_80k_isaid.py │ │ │ ├── fcn_hr18_512x512_80k_ade20k.py │ │ │ ├── fcn_hr18_512x512_80k_loveda.py │ │ │ ├── fcn_hr18_512x512_80k_potsdam.py │ │ │ ├── fcn_hr18_512x512_160k_ade20k.py │ │ │ ├── fcn_hr18_4x4_512x512_80k_vaihingen.py │ │ │ ├── fcn_hr18_512x512_20k_voc12aug.py │ │ │ └── fcn_hr18_512x512_40k_voc12aug.py │ │ ├── bisenetv1 │ │ │ └── bisenetv1_r18-d32_in1k-pre_4x8_1024x1024_160k_cityscapes.py │ │ ├── ocrnet │ │ │ ├── ocrnet_hr18_512x1024_160k_cityscapes.py │ │ │ ├── ocrnet_hr18_512x1024_40k_cityscapes.py │ │ │ └── ocrnet_hr18_512x1024_80k_cityscapes.py │ │ ├── fastfcn │ │ │ ├── fastfcn_r50-d32_jpu_aspp_4x4_512x1024_80k_cityscapes.py │ │ │ ├── fastfcn_r50-d32_jpu_enc_4x4_512x1024_80k_cityscapes.py │ │ │ └── fastfcn_r50-d32_jpu_psp_512x1024_80k_cityscapes.py │ │ ├── vit │ │ │ ├── upernet_deit-b16_mln_512x512_160k_ade20k.py │ │ │ ├── upernet_deit-b16_512x512_160k_ade20k.py │ │ │ ├── upernet_deit-b16_512x512_80k_ade20k.py │ │ │ └── upernet_deit-b16_ln_mln_512x512_160k_ade20k.py │ │ ├── bisenetv2 │ │ │ └── bisenetv2_fcn_fp16_4x4_1024x1024_160k_cityscapes.py │ │ ├── swin │ │ │ ├── upernet_swin_base_patch4_window7_512x512_160k_ade20k_pretrain_224x224_22K.py │ │ │ └── upernet_swin_base_patch4_window12_512x512_160k_ade20k_pretrain_384x384_22K.py │ │ └── erfnet │ │ │ └── erfnet_fcn_4x4_512x1024_160k_cityscapes.py │ ├── MANIFEST.in │ ├── docs │ │ ├── en │ │ │ ├── switch_language.md │ │ │ ├── _static │ │ │ │ └── css │ │ │ │ │ └── readthedocs.css │ │ │ └── tutorials │ │ │ │ └── index.rst │ │ └── zh_cn │ │ │ ├── switch_language.md │ │ │ ├── _static │ │ │ └── css │ │ │ │ └── readthedocs.css │ │ │ └── tutorials │ │ │ └── index.rst │ ├── .readthedocs.yml │ ├── docker │ │ └── serve │ │ │ ├── config.properties │ │ │ └── entrypoint.sh │ └── tools │ │ └── dist_train.sh └── mmtracking │ ├── mmtrack │ ├── .mim │ │ ├── configs │ │ ├── tools │ │ └── model-index.yml │ ├── core │ │ ├── motion │ │ │ └── __init__.py │ │ ├── hook │ │ │ └── __init__.py │ │ ├── anchor │ │ │ └── __init__.py │ │ └── bbox │ │ │ └── iou_calculators │ │ │ └── __init__.py │ ├── datasets │ │ └── parsers │ │ │ └── __init__.py │ ├── models │ │ ├── backbones │ │ │ └── __init__.py │ │ ├── roi_heads │ │ │ ├── bbox_heads │ │ │ │ └── __init__.py │ │ │ └── roi_extractors │ │ │ │ └── __init__.py │ │ ├── sot │ │ │ └── __init__.py │ │ ├── aggregators │ │ │ └── __init__.py │ │ ├── vid │ │ │ └── __init__.py │ │ └── losses │ │ │ └── __init__.py │ └── utils │ │ └── __init__.py │ ├── requirements │ ├── build.txt │ ├── readthedocs.txt │ ├── mminstall.txt │ ├── docs.txt │ ├── runtime.txt │ └── tests.txt │ ├── demo │ ├── demo.mp4 │ └── README.md │ ├── requirements.txt │ ├── configs │ ├── vid │ │ └── selsa │ │ │ └── selsa_faster_rcnn_r50_dc5_fp16_1x_imagenetvid.py │ ├── sot │ │ └── siamese_rpn │ │ │ └── siamese_rpn_r50_fp16_20e_lasot.py │ └── mot │ │ └── bytetrack │ │ └── bytetrack_yolox_x_crowdhuman_mot17-private.py │ ├── docs │ ├── en │ │ ├── switch_language.md │ │ ├── _static │ │ │ └── css │ │ │ │ └── readthedocs.css │ │ └── tutorials │ │ │ └── index.rst │ └── zh_cn │ │ ├── switch_language.md │ │ ├── _static │ │ └── css │ │ │ └── readthedocs.css │ │ └── tutorials │ │ └── index.rst │ ├── MANIFEST.in │ └── tests │ ├── test_data │ └── test_datasets │ │ └── __init__.py │ └── test_models │ └── test_forward │ └── __init__.py ├── figs └── overview.png ├── tools ├── data_converter │ └── __init__.py ├── debug.sh ├── dist_eval_loss.sh └── test.sh └── configs └── _base_ └── models └── paconv_cuda_ssg.py /det3d/ops/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /det3d/models/backbones/DCNv2/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /software/mmdet/mmdet/.mim/demo: -------------------------------------------------------------------------------- 1 | ../../demo -------------------------------------------------------------------------------- /software/mmdet/mmdet/.mim/tools: -------------------------------------------------------------------------------- 1 | ../../tools -------------------------------------------------------------------------------- /software/mmdet3d/requirements/build.txt: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /software/mmseg/mmseg/.mim/tools: -------------------------------------------------------------------------------- 1 | ../../tools -------------------------------------------------------------------------------- /det3d/models/backbones/DCNv2_t18/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /software/mmdet/mmdet/.mim/configs: -------------------------------------------------------------------------------- 1 | ../../configs -------------------------------------------------------------------------------- /software/mmseg/mmseg/.mim/configs: -------------------------------------------------------------------------------- 1 | ../../configs -------------------------------------------------------------------------------- /det3d/version.py: -------------------------------------------------------------------------------- 1 | __version__ = "0.1.0+9647dac" 2 | -------------------------------------------------------------------------------- /software/mmdet/docs/zh_cn/useful_tools.md: -------------------------------------------------------------------------------- 1 | ## 日志分析 2 | -------------------------------------------------------------------------------- /software/mmdet3d/docs/zh_cn/changelog.md: -------------------------------------------------------------------------------- 1 | # 变更日志 2 | -------------------------------------------------------------------------------- /software/mmtracking/mmtrack/.mim/configs: -------------------------------------------------------------------------------- 1 | ../../configs -------------------------------------------------------------------------------- /software/mmtracking/mmtrack/.mim/tools: -------------------------------------------------------------------------------- 1 | ../../tools -------------------------------------------------------------------------------- /software/mmdet3d/docs/zh_cn/compatibility.md: -------------------------------------------------------------------------------- 1 | ## 0.16.0 2 | -------------------------------------------------------------------------------- /software/mmdet/mmdet/.mim/model-index.yml: -------------------------------------------------------------------------------- 1 | ../../model-index.yml -------------------------------------------------------------------------------- /software/mmdet/requirements/mminstall.txt: -------------------------------------------------------------------------------- 1 | mmcv-full>=1.3.17 2 | -------------------------------------------------------------------------------- /software/mmseg/mmseg/.mim/model-index.yml: -------------------------------------------------------------------------------- 1 | ../../model-index.yml -------------------------------------------------------------------------------- /software/mmseg/requirements/optional.txt: -------------------------------------------------------------------------------- 1 | cityscapesscripts 2 | -------------------------------------------------------------------------------- /software/mmtracking/mmtrack/.mim/model-index.yml: -------------------------------------------------------------------------------- 1 | ../../model-index.yml -------------------------------------------------------------------------------- /software/mmdet/docs/zh_cn/tutorials/customize_runtime.md: -------------------------------------------------------------------------------- 1 | # 教程 5: 自定义训练配置 2 | -------------------------------------------------------------------------------- /software/mmdet/requirements/readthedocs.txt: -------------------------------------------------------------------------------- 1 | mmcv 2 | torch 3 | torchvision 4 | -------------------------------------------------------------------------------- /software/mmdet3d/.github/ISSUE_TEMPLATE/config.yml: -------------------------------------------------------------------------------- 1 | blank_issues_enabled: false 2 | -------------------------------------------------------------------------------- /software/mmtracking/requirements/build.txt: -------------------------------------------------------------------------------- 1 | cython 2 | numba==0.53.0 3 | numpy 4 | -------------------------------------------------------------------------------- /figs/overview.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenRobotLab/DORT/HEAD/figs/overview.png -------------------------------------------------------------------------------- /software/mmseg/requirements/mminstall.txt: -------------------------------------------------------------------------------- 1 | mmcls>=0.20.1 2 | mmcv-full>=1.4.4,<=1.5.0 3 | -------------------------------------------------------------------------------- /software/mmseg/tests/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) OpenMMLab. All rights reserved. 2 | -------------------------------------------------------------------------------- /tools/data_converter/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) OpenMMLab. All rights reserved. 2 | -------------------------------------------------------------------------------- /det3d/ops/bev_pool_v2/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) Phigent Robotics. All rights reserved. 2 | -------------------------------------------------------------------------------- /software/mmdet/requirements/albu.txt: -------------------------------------------------------------------------------- 1 | albumentations>=0.3.2 --no-binary qudida,albumentations 2 | -------------------------------------------------------------------------------- /software/mmseg/requirements/readthedocs.txt: -------------------------------------------------------------------------------- 1 | mmcv 2 | prettytable 3 | torch 4 | torchvision 5 | -------------------------------------------------------------------------------- /software/mmseg/tests/test_models/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) OpenMMLab. All rights reserved. 2 | -------------------------------------------------------------------------------- /software/mmdet3d/requirements/optional.txt: -------------------------------------------------------------------------------- 1 | open3d 2 | spconv 3 | waymo-open-dataset-tf-2-1-0==1.2.0 4 | -------------------------------------------------------------------------------- /software/mmdet3d/tools/data_converter/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) OpenMMLab. All rights reserved. 2 | -------------------------------------------------------------------------------- /software/mmtracking/requirements/readthedocs.txt: -------------------------------------------------------------------------------- 1 | mmcls 2 | mmcv 3 | mmdet 4 | torch 5 | torchvision 6 | -------------------------------------------------------------------------------- /tools/debug.sh: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env bash 2 | 3 | PORT=`expr $RANDOM % 40000 + 1000` 4 | echo $PORT 5 | 6 | -------------------------------------------------------------------------------- /det3d/apis/__init__.py: -------------------------------------------------------------------------------- 1 | from .test import single_gpu_get_loss 2 | 3 | __all__ = ['single_gpu_get_loss'] 4 | -------------------------------------------------------------------------------- /software/mmdet/requirements/runtime.txt: -------------------------------------------------------------------------------- 1 | matplotlib 2 | numpy 3 | pycocotools 4 | six 5 | terminaltables 6 | -------------------------------------------------------------------------------- /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_utils/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) OpenMMLab. All rights reserved. 2 | -------------------------------------------------------------------------------- /software/mmdet/demo/demo.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenRobotLab/DORT/HEAD/software/mmdet/demo/demo.mp4 -------------------------------------------------------------------------------- /software/mmseg/requirements/runtime.txt: -------------------------------------------------------------------------------- 1 | matplotlib 2 | mmcls>=0.20.1 3 | numpy 4 | packaging 5 | prettytable 6 | -------------------------------------------------------------------------------- /software/mmseg/tests/test_models/test_segmentors/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) OpenMMLab. All rights reserved. 2 | -------------------------------------------------------------------------------- /software/mmdet/requirements/build.txt: -------------------------------------------------------------------------------- 1 | # These must be installed before building mmdetection 2 | cython 3 | numpy 4 | -------------------------------------------------------------------------------- /software/mmdet/requirements/optional.txt: -------------------------------------------------------------------------------- 1 | cityscapesscripts 2 | imagecorruptions 3 | scipy 4 | sklearn 5 | timm 6 | -------------------------------------------------------------------------------- /software/mmseg/requirements/tests.txt: -------------------------------------------------------------------------------- 1 | codecov 2 | flake8 3 | interrogate 4 | pytest 5 | xdoctest>=0.10.0 6 | yapf 7 | -------------------------------------------------------------------------------- /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 | -------------------------------------------------------------------------------- /det3d/models/losses/__init__.py: -------------------------------------------------------------------------------- 1 | from .uncertainty_l1_loss import UncertaintyL1Loss 2 | 3 | __all__ = ['UncertaintyL1Loss'] -------------------------------------------------------------------------------- /software/mmtracking/demo/demo.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenRobotLab/DORT/HEAD/software/mmtracking/demo/demo.mp4 -------------------------------------------------------------------------------- /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/make.sh: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env bash 2 | rm *.so 3 | rm -r build/ 4 | python setup.py build develop 5 | -------------------------------------------------------------------------------- /det3d/models/motion/__init__.py: -------------------------------------------------------------------------------- 1 | from .ab3dmot_kalman_filter import AB3DMOTKalmanFilter 2 | 3 | __all__ = ['AB3DMOTKalmanFilter'] -------------------------------------------------------------------------------- /software/mmseg/requirements.txt: -------------------------------------------------------------------------------- 1 | -r requirements/optional.txt 2 | -r requirements/runtime.txt 3 | -r requirements/tests.txt 4 | -------------------------------------------------------------------------------- /software/mmseg/resources/seg_demo.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenRobotLab/DORT/HEAD/software/mmseg/resources/seg_demo.gif -------------------------------------------------------------------------------- /software/mmtracking/requirements.txt: -------------------------------------------------------------------------------- 1 | -r requirements/build.txt 2 | -r requirements/runtime.txt 3 | -r requirements/tests.txt 4 | -------------------------------------------------------------------------------- /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 -------------------------------------------------------------------------------- /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/readthedocs.txt: -------------------------------------------------------------------------------- 1 | mmcv>=1.4.8 2 | mmdet>=2.24.0 3 | mmsegmentation>=0.20.1 4 | torch 5 | torchvision 6 | -------------------------------------------------------------------------------- /det3d/core/bbox/assigners/__init__.py: -------------------------------------------------------------------------------- 1 | from .hungarian_assigner_3d import HungarianAssigner3D 2 | 3 | __all__ = ['HungarianAssigner3D'] 4 | -------------------------------------------------------------------------------- /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/mmdet3d/.dev_scripts/linter.sh: -------------------------------------------------------------------------------- 1 | yapf -r -i mmdet3d/ configs/ tests/ tools/ 2 | isort mmdet3d/ configs/ tests/ tools/ 3 | flake8 . 4 | -------------------------------------------------------------------------------- /software/mmdet3d/resources/nuimages_demo.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenRobotLab/DORT/HEAD/software/mmdet3d/resources/nuimages_demo.gif -------------------------------------------------------------------------------- /software/mmdet3d/resources/open3d_visual.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenRobotLab/DORT/HEAD/software/mmdet3d/resources/open3d_visual.gif -------------------------------------------------------------------------------- /det3d/models/track_heads/__init__.py: -------------------------------------------------------------------------------- 1 | from .quasi_dense_track_head import CustomQuasiDenseTrackHead 2 | 3 | __all__ = ['CustomQuasiDenseTrackHead'] -------------------------------------------------------------------------------- /det3d/core/visualizer/__init__.py: -------------------------------------------------------------------------------- 1 | # from .show_result import show_custom_multi_modality_result 2 | 3 | # __all__ = ['show_custom_multi_modality_result'] -------------------------------------------------------------------------------- /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/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/resources/mmdet3d_outdoor_demo.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenRobotLab/DORT/HEAD/software/mmdet3d/resources/mmdet3d_outdoor_demo.gif -------------------------------------------------------------------------------- /det3d/core/bbox/coders/__init__.py: -------------------------------------------------------------------------------- 1 | from .nms_free_coder import NMSFreeCoder 2 | from .nocs_coder import NOCSCoder 3 | __all__ = ['NMSFreeCoder', 'NOCSCoder'] 4 | -------------------------------------------------------------------------------- /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 | -------------------------------------------------------------------------------- /det3d/models/track_frameworks/__init__.py: -------------------------------------------------------------------------------- 1 | from .qdtrack import CustomQDTrack 2 | from .sort import CustomSORT 3 | 4 | __all__ = ['CustomQDTrack', 'CustomSORT'] 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/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/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/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/docs/zh_cn/tutorials/pytorch2onnx.md: -------------------------------------------------------------------------------- 1 | # 教程 8: Pytorch 到 ONNX 的模型转换（实验性支持） 2 | 3 | > ## [尝试使用新的 MMDeploy 來部署你的模型](https://mmdeploy.readthedocs.io/) 4 | -------------------------------------------------------------------------------- /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/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/mmdet3d/core/evaluation/waymo_utils/compute_detection_let_metrics_main: -------------------------------------------------------------------------------- 1 | /mnt/lustre/lianqing/data/waymo/eval_utils/compute_detection_let_metrics_main -------------------------------------------------------------------------------- /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_mstrain_3x_coco.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../common/mstrain_3x_coco.py', '../_base_/models/faster_rcnn_r50_fpn.py' 3 | ] 4 | -------------------------------------------------------------------------------- /software/mmdet3d/tests/test_samples/parta2_roihead_inputs.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenRobotLab/DORT/HEAD/software/mmdet3d/tests/test_samples/parta2_roihead_inputs.npz -------------------------------------------------------------------------------- /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/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/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/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/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/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/backbones/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) OpenMMLab. All rights reserved. 2 | from .sot_resnet import SOTResNet 3 | 4 | __all__ = ['SOTResNet'] 5 | -------------------------------------------------------------------------------- /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 | -------------------------------------------------------------------------------- /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_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/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/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/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 | -------------------------------------------------------------------------------- /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'] -------------------------------------------------------------------------------- /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/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/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/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/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/docs/en/switch_language.md: -------------------------------------------------------------------------------- 1 | ## English 2 | 3 | ## 简体中文 4 | -------------------------------------------------------------------------------- /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/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/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_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/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/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/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/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/docs/zh_cn/switch_language.md: -------------------------------------------------------------------------------- 1 | ## English 2 | 3 | ## 简体中文 4 | -------------------------------------------------------------------------------- /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/mmdet3d/docs/en/switch_language.md: -------------------------------------------------------------------------------- 1 | ## English 2 | 3 | ## 简体中文 4 | -------------------------------------------------------------------------------- /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/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/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_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/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_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/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/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/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/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_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_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_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/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/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/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_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/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/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_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/docs/en/switch_language.md: -------------------------------------------------------------------------------- 1 | ## English 2 | 3 | ## 简体中文 4 | -------------------------------------------------------------------------------- /software/mmtracking/docs/en/switch_language.md: -------------------------------------------------------------------------------- 1 | ## English 2 | 3 | ## 简体中文 4 | -------------------------------------------------------------------------------- /software/mmtracking/docs/zh_cn/switch_language.md: -------------------------------------------------------------------------------- 1 | ## English 2 | 3 | ## 简体中文 4 | -------------------------------------------------------------------------------- /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/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/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/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/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/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_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/docs/zh_cn/switch_language.md: -------------------------------------------------------------------------------- 1 | ## English 2 | 3 | ## 简体中文 4 | -------------------------------------------------------------------------------- /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/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/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/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/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_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/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_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/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/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_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/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/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_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/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_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/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/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/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/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_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/docs/zh_cn/switch_language.md: -------------------------------------------------------------------------------- 1 | ## English 2 | 3 | ## 简体中文 4 | -------------------------------------------------------------------------------- /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/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/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/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 | -------------------------------------------------------------------------------- /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 | -------------------------------------------------------------------------------- /det3d/models/trackers/__init__.py: -------------------------------------------------------------------------------- 1 | from .quasi_dense_tracker import CustomQuasiDenseTracker 2 | from .sort_tracker import CustomSORTTracker 3 | 4 | __all__ = ['CustomQuasiDenseTracker', 'CustomSORTTracker'] 5 | -------------------------------------------------------------------------------- /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_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/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_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/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/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/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/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/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/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_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/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_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_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/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/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_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/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/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/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_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/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_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_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_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_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/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/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/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/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/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/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_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/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/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/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/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/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/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/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/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/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_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/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/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/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_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_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/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/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_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/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_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_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/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/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/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/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/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/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/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/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/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/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/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_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/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_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/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_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_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_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/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/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/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/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/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/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/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/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/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/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/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_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/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/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_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_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/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/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/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/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/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/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/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/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_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_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/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_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/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/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/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 | -------------------------------------------------------------------------------- /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/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/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/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/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/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/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/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_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/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_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/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/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_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/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 | -------------------------------------------------------------------------------- /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/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/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_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/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_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/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/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_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/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_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/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_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/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/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/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/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/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/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/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/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/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/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/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/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/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/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/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/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/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_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/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/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/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_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_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 | -------------------------------------------------------------------------------- /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'] -------------------------------------------------------------------------------- /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/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/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/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/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/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/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/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/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/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/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/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/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/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/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/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/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/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/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/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_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/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/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/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/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/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/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/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/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/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/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/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/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/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/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/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/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/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/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/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/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/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/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/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/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/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/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/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/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/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/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/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/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/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/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/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/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/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_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/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/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/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/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/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/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/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/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/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/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/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/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/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/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_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/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/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/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_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/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/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/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_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/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/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/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/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/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/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/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/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/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_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/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/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/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/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/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/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/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/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/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/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/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/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/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/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/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/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_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/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/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/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/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/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/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/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/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/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/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/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/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/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/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/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/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/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/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/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/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/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/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/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/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/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/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/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/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/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/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/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/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/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/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/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/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/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/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/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/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/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/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/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/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/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/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/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/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/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/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/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/mmseg/configs/ann/ann_r50-d8_512x512_160k_ade20k.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/ann_r50-d8.py', '../_base_/datasets/ade20k.py', 3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_160k.py' 4 | ] 5 | model = dict( 6 | decode_head=dict(num_classes=150), auxiliary_head=dict(num_classes=150)) 7 | -------------------------------------------------------------------------------- /software/mmseg/configs/ann/ann_r50-d8_512x512_80k_ade20k.py: -------------------------------------------------------------------------------- 1 | _base_ = [ 2 | '../_base_/models/ann_r50-d8.py', '../_base_/datasets/ade20k.py', 3 | '../_base_/default_runtime.py', '../_base_/schedules/schedule_80k.py' 4 | ] 5 | model = dict( 6 | decode_head=dict(num_classes=150), auxiliary_head=dict(num_classes=150)) 7 | --------------------------------------------------------------------------------