├── .gitignore ├── LICENSE ├── MODEL_ZOO.md ├── README.md ├── adet ├── __init__.py ├── checkpoint │ ├── __init__.py │ └── adet_checkpoint.py ├── config │ ├── __init__.py │ ├── config.py │ └── defaults.py ├── data │ ├── __init__.py │ ├── augmentation.py │ ├── builtin.py │ ├── dataset_language.py │ ├── dataset_mapper.py │ ├── datasets │ │ └── text.py │ └── detection_utils.py ├── evaluation │ ├── __init__.py │ ├── rrc_evaluation_funcs.py │ ├── text_eval_script.py │ └── text_evaluation.py ├── layers │ ├── __init__.py │ ├── bezier_align.py │ ├── conv_with_kaiming_uniform.py │ ├── csrc │ │ ├── BezierAlign │ │ │ ├── BezierAlign.h │ │ │ ├── BezierAlign_cpu.cpp │ │ │ └── BezierAlign_cuda.cu │ │ ├── DefROIAlign │ │ │ ├── DefROIAlign.h │ │ │ └── DefROIAlign_cuda.cu │ │ ├── cuda_version.cu │ │ ├── ml_nms │ │ │ ├── ml_nms.cu │ │ │ └── ml_nms.h │ │ └── vision.cpp │ ├── def_roi_align.py │ ├── deform_conv.py │ ├── gcn.py │ ├── iou_loss.py │ ├── ml_nms.py │ └── naive_group_norm.py ├── modeling │ ├── MEInst │ │ ├── LME │ │ │ ├── MaskLoader.py │ │ │ ├── __init__.py │ │ │ ├── mask_evaluation.py │ │ │ ├── mask_generation.py │ │ │ └── utils.py │ │ ├── MEInst.py │ │ ├── MEInst_outputs.py │ │ ├── MaskEncoding.py │ │ └── __init__.py │ ├── __init__.py │ ├── abinet │ │ ├── attention.py │ │ ├── attn_text_head.py │ │ ├── feat_seq_modeling.py │ │ ├── model.py │ │ ├── model_abinet.py │ │ ├── model_abinet_iter.py │ │ ├── model_alignment.py │ │ ├── model_language.py │ │ ├── model_vision.py │ │ └── transformer.py │ ├── backbone │ │ ├── __init__.py │ │ ├── bifpn.py │ │ ├── dla.py │ │ ├── fpn.py │ │ ├── lpf.py │ │ ├── mobilenet.py │ │ ├── resnet_interval.py │ │ ├── resnet_lpf.py │ │ └── vovnet.py │ ├── batext │ │ ├── __init__.py │ │ ├── batext.py │ │ └── batext_outputs.py │ ├── blendmask │ │ ├── __init__.py │ │ ├── basis_module.py │ │ ├── blender.py │ │ └── blendmask.py │ ├── condinst │ │ ├── __init__.py │ │ ├── condinst.py │ │ ├── dynamic_mask_head.py │ │ └── mask_branch.py │ ├── fcos │ │ ├── __init__.py │ │ ├── fcos.py │ │ └── fcos_outputs.py │ ├── one_stage_detector.py │ ├── poolers.py │ ├── roi_heads │ │ ├── __init__.py │ │ ├── attn_predictor.py │ │ └── text_head.py │ └── solov2 │ │ ├── __init__.py │ │ ├── loss.py │ │ ├── solov2.py │ │ └── utils.py ├── structures │ ├── __init__.py │ └── beziers.py └── utils │ ├── __init__.py │ ├── comm.py │ ├── measures.py │ └── visualizer.py ├── configs ├── ABINet │ ├── Base-ABINet.yaml │ ├── CTW1500.yaml │ ├── ICDAR2015.yaml │ ├── Pretrain-chn.yaml │ ├── Pretrain.yaml │ ├── ReCTS.yaml │ └── TotalText.yaml ├── BAText │ ├── Base-BAText.yaml │ ├── CTW1500 │ │ ├── Base-CTW1500.yaml │ │ └── attn_R_50.yaml │ ├── Pretrain │ │ ├── Base-Pretrain.yaml │ │ └── attn_R_50.yaml │ ├── README.md │ └── TotalText │ │ ├── Base-TotalText.yaml │ │ └── attn_R_50.yaml ├── BlendMask │ ├── 550_R_50_1x.yaml │ ├── 550_R_50_3x.yaml │ ├── 550_R_50_dcni3_5x.yaml │ ├── Base-550.yaml │ ├── Base-BlendMask.yaml │ ├── Base-RT.yaml │ ├── DLA_34_syncbn_4x.yaml │ ├── Panoptic │ │ ├── Base-Panoptic.yaml │ │ ├── R_101_3x.yaml │ │ ├── R_101_dcni3_5x.yaml │ │ ├── R_50_1x.yaml │ │ ├── R_50_3x.yaml │ │ └── R_50_dcni3_5x.yaml │ ├── Person │ │ ├── Base-Person.yaml │ │ └── R_50_1x.yaml │ ├── README.md │ ├── RT_R_50_4x_bn-head_syncbn_shtw.yaml │ ├── RT_R_50_4x_syncbn_shtw.yaml │ ├── R_101_3x.yaml │ ├── R_101_dcni3_5x.yaml │ ├── R_50_1x.yaml │ └── R_50_3x.yaml ├── BoxInst │ ├── Base-BoxInst.yaml │ ├── MS_R_101_1x.yaml │ ├── MS_R_101_3x.yaml │ ├── MS_R_101_BiFPN_3x.yaml │ ├── MS_R_101_BiFPN_dcni3_3x.yaml │ ├── MS_R_50_1x.yaml │ ├── MS_R_50_3x.yaml │ ├── MS_R_50_BiFPN_1x.yaml │ ├── MS_R_50_BiFPN_3x.yaml │ └── README.md ├── CondInst │ ├── Base-CondInst.yaml │ ├── MS_R_101_1x.yaml │ ├── MS_R_101_3x.yaml │ ├── MS_R_101_3x_sem.yaml │ ├── MS_R_101_BiFPN_3x.yaml │ ├── MS_R_101_BiFPN_3x_sem.yaml │ ├── MS_R_50_1x.yaml │ ├── MS_R_50_3x.yaml │ ├── MS_R_50_3x_sem.yaml │ ├── MS_R_50_BiFPN_1x.yaml │ ├── MS_R_50_BiFPN_3x.yaml │ ├── MS_R_50_BiFPN_3x_sem.yaml │ └── README.md ├── FCOS-Detection │ ├── Base-FCOS.yaml │ ├── FCOS_RT │ │ ├── MS_DLA_34_4x_syncbn.yaml │ │ ├── MS_DLA_34_4x_syncbn_bn_head.yaml │ │ ├── MS_DLA_34_4x_syncbn_shared_towers.yaml │ │ ├── MS_DLA_34_4x_syncbn_shared_towers_bn_head.yaml │ │ ├── MS_R_50_4x_syncbn.yaml │ │ └── MS_R_50_4x_syncbn_bn_head.yaml │ ├── MS_R_101_2x.yaml │ ├── MS_R_50_2x.yaml │ ├── MS_X_101_32x8d_2x.yaml │ ├── MS_X_101_32x8d_2x_dcnv2.yaml │ ├── MS_X_101_64x4d_2x.yaml │ ├── MS_X_101_64x4d_2x_dcnv2.yaml │ ├── README.md │ ├── R_50_1x.yaml │ └── vovnet │ │ ├── MS_V_39_3x.yaml │ │ ├── MS_V_57_3x.yaml │ │ ├── MS_V_99_3x.yaml │ │ └── README.md ├── MEInst-InstanceSegmentation │ ├── Base-MEInst.yaml │ ├── MEInst_R_50_1x.yaml │ ├── MEInst_R_50_1x_none.yaml │ ├── MEInst_R_50_3x.yaml │ ├── MEInst_R_50_3x_512.yaml │ └── README.md ├── RCNN │ ├── 550_R_50_FPN_3x.yaml │ ├── Base-RCNN.yaml │ ├── LVIS │ │ └── R_50_1x.yaml │ └── R_101_3x.yaml └── SOLOv2 │ ├── Base-SOLOv2.yaml │ ├── R101_3x.yaml │ ├── R50_3x.yaml │ └── README.md ├── datasets ├── README.md ├── gen_coco_person.py ├── prepare_thing_sem_from_instance.py └── prepare_thing_sem_from_lvis.py ├── demo ├── demo.py └── predictor.py ├── docker └── Dockerfile ├── docs ├── .gitignore ├── Makefile ├── adel-logo.svg ├── conf.py ├── index.rst ├── modules │ ├── checkpoint.rst │ ├── config.rst │ ├── data.rst │ ├── index.rst │ ├── layers.rst │ ├── modeling.rst │ └── utils.rst └── requirements.txt ├── figs └── framework.png ├── onnx ├── .gitignore ├── README.md ├── export_model_to_onnx.py ├── pytorch-onnx-caffe-ncnn-rt.sh ├── pytorch-onnx-caffe-ncnn.sh └── test_onnxruntime.py ├── setup.py └── tools ├── compute_flops.py ├── convert_fcos_weight.py ├── remove_optim_from_ckpt.py ├── rename_blendmask.py ├── train_net.py └── visualize_data.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/LICENSE -------------------------------------------------------------------------------- /MODEL_ZOO.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/MODEL_ZOO.md -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/README.md -------------------------------------------------------------------------------- /adet/__init__.py: -------------------------------------------------------------------------------- 1 | from adet import modeling 2 | 3 | __version__ = "0.1.1" 4 | -------------------------------------------------------------------------------- /adet/checkpoint/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/checkpoint/__init__.py -------------------------------------------------------------------------------- /adet/checkpoint/adet_checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/checkpoint/adet_checkpoint.py -------------------------------------------------------------------------------- /adet/config/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/config/__init__.py -------------------------------------------------------------------------------- /adet/config/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/config/config.py -------------------------------------------------------------------------------- /adet/config/defaults.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/config/defaults.py -------------------------------------------------------------------------------- /adet/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/data/__init__.py -------------------------------------------------------------------------------- /adet/data/augmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/data/augmentation.py -------------------------------------------------------------------------------- /adet/data/builtin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/data/builtin.py -------------------------------------------------------------------------------- /adet/data/dataset_language.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/data/dataset_language.py -------------------------------------------------------------------------------- /adet/data/dataset_mapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/data/dataset_mapper.py -------------------------------------------------------------------------------- /adet/data/datasets/text.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/data/datasets/text.py -------------------------------------------------------------------------------- /adet/data/detection_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/data/detection_utils.py -------------------------------------------------------------------------------- /adet/evaluation/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/evaluation/__init__.py -------------------------------------------------------------------------------- /adet/evaluation/rrc_evaluation_funcs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/evaluation/rrc_evaluation_funcs.py -------------------------------------------------------------------------------- /adet/evaluation/text_eval_script.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/evaluation/text_eval_script.py -------------------------------------------------------------------------------- /adet/evaluation/text_evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/evaluation/text_evaluation.py -------------------------------------------------------------------------------- /adet/layers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/layers/__init__.py -------------------------------------------------------------------------------- /adet/layers/bezier_align.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/layers/bezier_align.py -------------------------------------------------------------------------------- /adet/layers/conv_with_kaiming_uniform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/layers/conv_with_kaiming_uniform.py -------------------------------------------------------------------------------- /adet/layers/csrc/BezierAlign/BezierAlign.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/layers/csrc/BezierAlign/BezierAlign.h -------------------------------------------------------------------------------- /adet/layers/csrc/BezierAlign/BezierAlign_cpu.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/layers/csrc/BezierAlign/BezierAlign_cpu.cpp -------------------------------------------------------------------------------- /adet/layers/csrc/BezierAlign/BezierAlign_cuda.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/layers/csrc/BezierAlign/BezierAlign_cuda.cu -------------------------------------------------------------------------------- /adet/layers/csrc/DefROIAlign/DefROIAlign.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/layers/csrc/DefROIAlign/DefROIAlign.h -------------------------------------------------------------------------------- /adet/layers/csrc/DefROIAlign/DefROIAlign_cuda.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/layers/csrc/DefROIAlign/DefROIAlign_cuda.cu -------------------------------------------------------------------------------- /adet/layers/csrc/cuda_version.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/layers/csrc/cuda_version.cu -------------------------------------------------------------------------------- /adet/layers/csrc/ml_nms/ml_nms.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/layers/csrc/ml_nms/ml_nms.cu -------------------------------------------------------------------------------- /adet/layers/csrc/ml_nms/ml_nms.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/layers/csrc/ml_nms/ml_nms.h -------------------------------------------------------------------------------- /adet/layers/csrc/vision.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/layers/csrc/vision.cpp -------------------------------------------------------------------------------- /adet/layers/def_roi_align.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/layers/def_roi_align.py -------------------------------------------------------------------------------- /adet/layers/deform_conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/layers/deform_conv.py -------------------------------------------------------------------------------- /adet/layers/gcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/layers/gcn.py -------------------------------------------------------------------------------- /adet/layers/iou_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/layers/iou_loss.py -------------------------------------------------------------------------------- /adet/layers/ml_nms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/layers/ml_nms.py -------------------------------------------------------------------------------- /adet/layers/naive_group_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/layers/naive_group_norm.py -------------------------------------------------------------------------------- /adet/modeling/MEInst/LME/MaskLoader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/modeling/MEInst/LME/MaskLoader.py -------------------------------------------------------------------------------- /adet/modeling/MEInst/LME/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/modeling/MEInst/LME/__init__.py -------------------------------------------------------------------------------- /adet/modeling/MEInst/LME/mask_evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/modeling/MEInst/LME/mask_evaluation.py -------------------------------------------------------------------------------- /adet/modeling/MEInst/LME/mask_generation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/modeling/MEInst/LME/mask_generation.py -------------------------------------------------------------------------------- /adet/modeling/MEInst/LME/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/modeling/MEInst/LME/utils.py -------------------------------------------------------------------------------- /adet/modeling/MEInst/MEInst.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/modeling/MEInst/MEInst.py -------------------------------------------------------------------------------- /adet/modeling/MEInst/MEInst_outputs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/modeling/MEInst/MEInst_outputs.py -------------------------------------------------------------------------------- /adet/modeling/MEInst/MaskEncoding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/modeling/MEInst/MaskEncoding.py -------------------------------------------------------------------------------- /adet/modeling/MEInst/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/modeling/MEInst/__init__.py -------------------------------------------------------------------------------- /adet/modeling/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/modeling/__init__.py -------------------------------------------------------------------------------- /adet/modeling/abinet/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/modeling/abinet/attention.py -------------------------------------------------------------------------------- /adet/modeling/abinet/attn_text_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/modeling/abinet/attn_text_head.py -------------------------------------------------------------------------------- /adet/modeling/abinet/feat_seq_modeling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/modeling/abinet/feat_seq_modeling.py -------------------------------------------------------------------------------- /adet/modeling/abinet/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/modeling/abinet/model.py -------------------------------------------------------------------------------- /adet/modeling/abinet/model_abinet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/modeling/abinet/model_abinet.py -------------------------------------------------------------------------------- /adet/modeling/abinet/model_abinet_iter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/modeling/abinet/model_abinet_iter.py -------------------------------------------------------------------------------- /adet/modeling/abinet/model_alignment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/modeling/abinet/model_alignment.py -------------------------------------------------------------------------------- /adet/modeling/abinet/model_language.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/modeling/abinet/model_language.py -------------------------------------------------------------------------------- /adet/modeling/abinet/model_vision.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/modeling/abinet/model_vision.py -------------------------------------------------------------------------------- /adet/modeling/abinet/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/modeling/abinet/transformer.py -------------------------------------------------------------------------------- /adet/modeling/backbone/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/modeling/backbone/__init__.py -------------------------------------------------------------------------------- /adet/modeling/backbone/bifpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/modeling/backbone/bifpn.py -------------------------------------------------------------------------------- /adet/modeling/backbone/dla.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/modeling/backbone/dla.py -------------------------------------------------------------------------------- /adet/modeling/backbone/fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/modeling/backbone/fpn.py -------------------------------------------------------------------------------- /adet/modeling/backbone/lpf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/modeling/backbone/lpf.py -------------------------------------------------------------------------------- /adet/modeling/backbone/mobilenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/modeling/backbone/mobilenet.py -------------------------------------------------------------------------------- /adet/modeling/backbone/resnet_interval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/modeling/backbone/resnet_interval.py -------------------------------------------------------------------------------- /adet/modeling/backbone/resnet_lpf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/modeling/backbone/resnet_lpf.py -------------------------------------------------------------------------------- /adet/modeling/backbone/vovnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/modeling/backbone/vovnet.py -------------------------------------------------------------------------------- /adet/modeling/batext/__init__.py: -------------------------------------------------------------------------------- 1 | from .batext import BAText 2 | -------------------------------------------------------------------------------- /adet/modeling/batext/batext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/modeling/batext/batext.py -------------------------------------------------------------------------------- /adet/modeling/batext/batext_outputs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/modeling/batext/batext_outputs.py -------------------------------------------------------------------------------- /adet/modeling/blendmask/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/modeling/blendmask/__init__.py -------------------------------------------------------------------------------- /adet/modeling/blendmask/basis_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/modeling/blendmask/basis_module.py -------------------------------------------------------------------------------- /adet/modeling/blendmask/blender.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/modeling/blendmask/blender.py -------------------------------------------------------------------------------- /adet/modeling/blendmask/blendmask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/modeling/blendmask/blendmask.py -------------------------------------------------------------------------------- /adet/modeling/condinst/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/modeling/condinst/__init__.py -------------------------------------------------------------------------------- /adet/modeling/condinst/condinst.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/modeling/condinst/condinst.py -------------------------------------------------------------------------------- /adet/modeling/condinst/dynamic_mask_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/modeling/condinst/dynamic_mask_head.py -------------------------------------------------------------------------------- /adet/modeling/condinst/mask_branch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/modeling/condinst/mask_branch.py -------------------------------------------------------------------------------- /adet/modeling/fcos/__init__.py: -------------------------------------------------------------------------------- 1 | from .fcos import FCOS 2 | -------------------------------------------------------------------------------- /adet/modeling/fcos/fcos.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/modeling/fcos/fcos.py -------------------------------------------------------------------------------- /adet/modeling/fcos/fcos_outputs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/modeling/fcos/fcos_outputs.py -------------------------------------------------------------------------------- /adet/modeling/one_stage_detector.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/modeling/one_stage_detector.py -------------------------------------------------------------------------------- /adet/modeling/poolers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/modeling/poolers.py -------------------------------------------------------------------------------- /adet/modeling/roi_heads/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /adet/modeling/roi_heads/attn_predictor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/modeling/roi_heads/attn_predictor.py -------------------------------------------------------------------------------- /adet/modeling/roi_heads/text_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/modeling/roi_heads/text_head.py -------------------------------------------------------------------------------- /adet/modeling/solov2/__init__.py: -------------------------------------------------------------------------------- 1 | from .solov2 import SOLOv2 2 | -------------------------------------------------------------------------------- /adet/modeling/solov2/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/modeling/solov2/loss.py -------------------------------------------------------------------------------- /adet/modeling/solov2/solov2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/modeling/solov2/solov2.py -------------------------------------------------------------------------------- /adet/modeling/solov2/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/modeling/solov2/utils.py -------------------------------------------------------------------------------- /adet/structures/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/structures/__init__.py -------------------------------------------------------------------------------- /adet/structures/beziers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/structures/beziers.py -------------------------------------------------------------------------------- /adet/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /adet/utils/comm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/utils/comm.py -------------------------------------------------------------------------------- /adet/utils/measures.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/utils/measures.py -------------------------------------------------------------------------------- /adet/utils/visualizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/adet/utils/visualizer.py -------------------------------------------------------------------------------- /configs/ABINet/Base-ABINet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/ABINet/Base-ABINet.yaml -------------------------------------------------------------------------------- /configs/ABINet/CTW1500.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/ABINet/CTW1500.yaml -------------------------------------------------------------------------------- /configs/ABINet/ICDAR2015.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/ABINet/ICDAR2015.yaml -------------------------------------------------------------------------------- /configs/ABINet/Pretrain-chn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/ABINet/Pretrain-chn.yaml -------------------------------------------------------------------------------- /configs/ABINet/Pretrain.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/ABINet/Pretrain.yaml -------------------------------------------------------------------------------- /configs/ABINet/ReCTS.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/ABINet/ReCTS.yaml -------------------------------------------------------------------------------- /configs/ABINet/TotalText.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/ABINet/TotalText.yaml -------------------------------------------------------------------------------- /configs/BAText/Base-BAText.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/BAText/Base-BAText.yaml -------------------------------------------------------------------------------- /configs/BAText/CTW1500/Base-CTW1500.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/BAText/CTW1500/Base-CTW1500.yaml -------------------------------------------------------------------------------- /configs/BAText/CTW1500/attn_R_50.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/BAText/CTW1500/attn_R_50.yaml -------------------------------------------------------------------------------- /configs/BAText/Pretrain/Base-Pretrain.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/BAText/Pretrain/Base-Pretrain.yaml -------------------------------------------------------------------------------- /configs/BAText/Pretrain/attn_R_50.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/BAText/Pretrain/attn_R_50.yaml -------------------------------------------------------------------------------- /configs/BAText/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/BAText/README.md -------------------------------------------------------------------------------- /configs/BAText/TotalText/Base-TotalText.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/BAText/TotalText/Base-TotalText.yaml -------------------------------------------------------------------------------- /configs/BAText/TotalText/attn_R_50.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/BAText/TotalText/attn_R_50.yaml -------------------------------------------------------------------------------- /configs/BlendMask/550_R_50_1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/BlendMask/550_R_50_1x.yaml -------------------------------------------------------------------------------- /configs/BlendMask/550_R_50_3x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/BlendMask/550_R_50_3x.yaml -------------------------------------------------------------------------------- /configs/BlendMask/550_R_50_dcni3_5x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/BlendMask/550_R_50_dcni3_5x.yaml -------------------------------------------------------------------------------- /configs/BlendMask/Base-550.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/BlendMask/Base-550.yaml -------------------------------------------------------------------------------- /configs/BlendMask/Base-BlendMask.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/BlendMask/Base-BlendMask.yaml -------------------------------------------------------------------------------- /configs/BlendMask/Base-RT.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/BlendMask/Base-RT.yaml -------------------------------------------------------------------------------- /configs/BlendMask/DLA_34_syncbn_4x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/BlendMask/DLA_34_syncbn_4x.yaml -------------------------------------------------------------------------------- /configs/BlendMask/Panoptic/Base-Panoptic.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/BlendMask/Panoptic/Base-Panoptic.yaml -------------------------------------------------------------------------------- /configs/BlendMask/Panoptic/R_101_3x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/BlendMask/Panoptic/R_101_3x.yaml -------------------------------------------------------------------------------- /configs/BlendMask/Panoptic/R_101_dcni3_5x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/BlendMask/Panoptic/R_101_dcni3_5x.yaml -------------------------------------------------------------------------------- /configs/BlendMask/Panoptic/R_50_1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/BlendMask/Panoptic/R_50_1x.yaml -------------------------------------------------------------------------------- /configs/BlendMask/Panoptic/R_50_3x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/BlendMask/Panoptic/R_50_3x.yaml -------------------------------------------------------------------------------- /configs/BlendMask/Panoptic/R_50_dcni3_5x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/BlendMask/Panoptic/R_50_dcni3_5x.yaml -------------------------------------------------------------------------------- /configs/BlendMask/Person/Base-Person.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/BlendMask/Person/Base-Person.yaml -------------------------------------------------------------------------------- /configs/BlendMask/Person/R_50_1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/BlendMask/Person/R_50_1x.yaml -------------------------------------------------------------------------------- /configs/BlendMask/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/BlendMask/README.md -------------------------------------------------------------------------------- /configs/BlendMask/RT_R_50_4x_bn-head_syncbn_shtw.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/BlendMask/RT_R_50_4x_bn-head_syncbn_shtw.yaml -------------------------------------------------------------------------------- /configs/BlendMask/RT_R_50_4x_syncbn_shtw.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/BlendMask/RT_R_50_4x_syncbn_shtw.yaml -------------------------------------------------------------------------------- /configs/BlendMask/R_101_3x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/BlendMask/R_101_3x.yaml -------------------------------------------------------------------------------- /configs/BlendMask/R_101_dcni3_5x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/BlendMask/R_101_dcni3_5x.yaml -------------------------------------------------------------------------------- /configs/BlendMask/R_50_1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/BlendMask/R_50_1x.yaml -------------------------------------------------------------------------------- /configs/BlendMask/R_50_3x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/BlendMask/R_50_3x.yaml -------------------------------------------------------------------------------- /configs/BoxInst/Base-BoxInst.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/BoxInst/Base-BoxInst.yaml -------------------------------------------------------------------------------- /configs/BoxInst/MS_R_101_1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/BoxInst/MS_R_101_1x.yaml -------------------------------------------------------------------------------- /configs/BoxInst/MS_R_101_3x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/BoxInst/MS_R_101_3x.yaml -------------------------------------------------------------------------------- /configs/BoxInst/MS_R_101_BiFPN_3x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/BoxInst/MS_R_101_BiFPN_3x.yaml -------------------------------------------------------------------------------- /configs/BoxInst/MS_R_101_BiFPN_dcni3_3x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/BoxInst/MS_R_101_BiFPN_dcni3_3x.yaml -------------------------------------------------------------------------------- /configs/BoxInst/MS_R_50_1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/BoxInst/MS_R_50_1x.yaml -------------------------------------------------------------------------------- /configs/BoxInst/MS_R_50_3x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/BoxInst/MS_R_50_3x.yaml -------------------------------------------------------------------------------- /configs/BoxInst/MS_R_50_BiFPN_1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/BoxInst/MS_R_50_BiFPN_1x.yaml -------------------------------------------------------------------------------- /configs/BoxInst/MS_R_50_BiFPN_3x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/BoxInst/MS_R_50_BiFPN_3x.yaml -------------------------------------------------------------------------------- /configs/BoxInst/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/BoxInst/README.md -------------------------------------------------------------------------------- /configs/CondInst/Base-CondInst.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/CondInst/Base-CondInst.yaml -------------------------------------------------------------------------------- /configs/CondInst/MS_R_101_1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/CondInst/MS_R_101_1x.yaml -------------------------------------------------------------------------------- /configs/CondInst/MS_R_101_3x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/CondInst/MS_R_101_3x.yaml -------------------------------------------------------------------------------- /configs/CondInst/MS_R_101_3x_sem.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/CondInst/MS_R_101_3x_sem.yaml -------------------------------------------------------------------------------- /configs/CondInst/MS_R_101_BiFPN_3x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/CondInst/MS_R_101_BiFPN_3x.yaml -------------------------------------------------------------------------------- /configs/CondInst/MS_R_101_BiFPN_3x_sem.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/CondInst/MS_R_101_BiFPN_3x_sem.yaml -------------------------------------------------------------------------------- /configs/CondInst/MS_R_50_1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/CondInst/MS_R_50_1x.yaml -------------------------------------------------------------------------------- /configs/CondInst/MS_R_50_3x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/CondInst/MS_R_50_3x.yaml -------------------------------------------------------------------------------- /configs/CondInst/MS_R_50_3x_sem.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/CondInst/MS_R_50_3x_sem.yaml -------------------------------------------------------------------------------- /configs/CondInst/MS_R_50_BiFPN_1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/CondInst/MS_R_50_BiFPN_1x.yaml -------------------------------------------------------------------------------- /configs/CondInst/MS_R_50_BiFPN_3x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/CondInst/MS_R_50_BiFPN_3x.yaml -------------------------------------------------------------------------------- /configs/CondInst/MS_R_50_BiFPN_3x_sem.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/CondInst/MS_R_50_BiFPN_3x_sem.yaml -------------------------------------------------------------------------------- /configs/CondInst/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/CondInst/README.md -------------------------------------------------------------------------------- /configs/FCOS-Detection/Base-FCOS.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/FCOS-Detection/Base-FCOS.yaml -------------------------------------------------------------------------------- /configs/FCOS-Detection/FCOS_RT/MS_DLA_34_4x_syncbn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/FCOS-Detection/FCOS_RT/MS_DLA_34_4x_syncbn.yaml -------------------------------------------------------------------------------- /configs/FCOS-Detection/FCOS_RT/MS_DLA_34_4x_syncbn_bn_head.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/FCOS-Detection/FCOS_RT/MS_DLA_34_4x_syncbn_bn_head.yaml -------------------------------------------------------------------------------- /configs/FCOS-Detection/FCOS_RT/MS_DLA_34_4x_syncbn_shared_towers.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/FCOS-Detection/FCOS_RT/MS_DLA_34_4x_syncbn_shared_towers.yaml -------------------------------------------------------------------------------- /configs/FCOS-Detection/FCOS_RT/MS_DLA_34_4x_syncbn_shared_towers_bn_head.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/FCOS-Detection/FCOS_RT/MS_DLA_34_4x_syncbn_shared_towers_bn_head.yaml -------------------------------------------------------------------------------- /configs/FCOS-Detection/FCOS_RT/MS_R_50_4x_syncbn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/FCOS-Detection/FCOS_RT/MS_R_50_4x_syncbn.yaml -------------------------------------------------------------------------------- /configs/FCOS-Detection/FCOS_RT/MS_R_50_4x_syncbn_bn_head.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/FCOS-Detection/FCOS_RT/MS_R_50_4x_syncbn_bn_head.yaml -------------------------------------------------------------------------------- /configs/FCOS-Detection/MS_R_101_2x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/FCOS-Detection/MS_R_101_2x.yaml -------------------------------------------------------------------------------- /configs/FCOS-Detection/MS_R_50_2x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/FCOS-Detection/MS_R_50_2x.yaml -------------------------------------------------------------------------------- /configs/FCOS-Detection/MS_X_101_32x8d_2x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/FCOS-Detection/MS_X_101_32x8d_2x.yaml -------------------------------------------------------------------------------- /configs/FCOS-Detection/MS_X_101_32x8d_2x_dcnv2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/FCOS-Detection/MS_X_101_32x8d_2x_dcnv2.yaml -------------------------------------------------------------------------------- /configs/FCOS-Detection/MS_X_101_64x4d_2x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/FCOS-Detection/MS_X_101_64x4d_2x.yaml -------------------------------------------------------------------------------- /configs/FCOS-Detection/MS_X_101_64x4d_2x_dcnv2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/FCOS-Detection/MS_X_101_64x4d_2x_dcnv2.yaml -------------------------------------------------------------------------------- /configs/FCOS-Detection/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/FCOS-Detection/README.md -------------------------------------------------------------------------------- /configs/FCOS-Detection/R_50_1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/FCOS-Detection/R_50_1x.yaml -------------------------------------------------------------------------------- /configs/FCOS-Detection/vovnet/MS_V_39_3x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/FCOS-Detection/vovnet/MS_V_39_3x.yaml -------------------------------------------------------------------------------- /configs/FCOS-Detection/vovnet/MS_V_57_3x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/FCOS-Detection/vovnet/MS_V_57_3x.yaml -------------------------------------------------------------------------------- /configs/FCOS-Detection/vovnet/MS_V_99_3x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/FCOS-Detection/vovnet/MS_V_99_3x.yaml -------------------------------------------------------------------------------- /configs/FCOS-Detection/vovnet/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/FCOS-Detection/vovnet/README.md -------------------------------------------------------------------------------- /configs/MEInst-InstanceSegmentation/Base-MEInst.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/MEInst-InstanceSegmentation/Base-MEInst.yaml -------------------------------------------------------------------------------- /configs/MEInst-InstanceSegmentation/MEInst_R_50_1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/MEInst-InstanceSegmentation/MEInst_R_50_1x.yaml -------------------------------------------------------------------------------- /configs/MEInst-InstanceSegmentation/MEInst_R_50_1x_none.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/MEInst-InstanceSegmentation/MEInst_R_50_1x_none.yaml -------------------------------------------------------------------------------- /configs/MEInst-InstanceSegmentation/MEInst_R_50_3x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/MEInst-InstanceSegmentation/MEInst_R_50_3x.yaml -------------------------------------------------------------------------------- /configs/MEInst-InstanceSegmentation/MEInst_R_50_3x_512.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/MEInst-InstanceSegmentation/MEInst_R_50_3x_512.yaml -------------------------------------------------------------------------------- /configs/MEInst-InstanceSegmentation/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/MEInst-InstanceSegmentation/README.md -------------------------------------------------------------------------------- /configs/RCNN/550_R_50_FPN_3x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/RCNN/550_R_50_FPN_3x.yaml -------------------------------------------------------------------------------- /configs/RCNN/Base-RCNN.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/RCNN/Base-RCNN.yaml -------------------------------------------------------------------------------- /configs/RCNN/LVIS/R_50_1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/RCNN/LVIS/R_50_1x.yaml -------------------------------------------------------------------------------- /configs/RCNN/R_101_3x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/RCNN/R_101_3x.yaml -------------------------------------------------------------------------------- /configs/SOLOv2/Base-SOLOv2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/SOLOv2/Base-SOLOv2.yaml -------------------------------------------------------------------------------- /configs/SOLOv2/R101_3x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/SOLOv2/R101_3x.yaml -------------------------------------------------------------------------------- /configs/SOLOv2/R50_3x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/SOLOv2/R50_3x.yaml -------------------------------------------------------------------------------- /configs/SOLOv2/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/configs/SOLOv2/README.md -------------------------------------------------------------------------------- /datasets/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/datasets/README.md -------------------------------------------------------------------------------- /datasets/gen_coco_person.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/datasets/gen_coco_person.py -------------------------------------------------------------------------------- /datasets/prepare_thing_sem_from_instance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/datasets/prepare_thing_sem_from_instance.py -------------------------------------------------------------------------------- /datasets/prepare_thing_sem_from_lvis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/datasets/prepare_thing_sem_from_lvis.py -------------------------------------------------------------------------------- /demo/demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/demo/demo.py -------------------------------------------------------------------------------- /demo/predictor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/demo/predictor.py -------------------------------------------------------------------------------- /docker/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/docker/Dockerfile -------------------------------------------------------------------------------- /docs/.gitignore: -------------------------------------------------------------------------------- 1 | _build -------------------------------------------------------------------------------- /docs/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/docs/Makefile -------------------------------------------------------------------------------- /docs/adel-logo.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/docs/adel-logo.svg -------------------------------------------------------------------------------- /docs/conf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/docs/conf.py -------------------------------------------------------------------------------- /docs/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/docs/index.rst -------------------------------------------------------------------------------- /docs/modules/checkpoint.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/docs/modules/checkpoint.rst -------------------------------------------------------------------------------- /docs/modules/config.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/docs/modules/config.rst -------------------------------------------------------------------------------- /docs/modules/data.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/docs/modules/data.rst -------------------------------------------------------------------------------- /docs/modules/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/docs/modules/index.rst -------------------------------------------------------------------------------- /docs/modules/layers.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/docs/modules/layers.rst -------------------------------------------------------------------------------- /docs/modules/modeling.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/docs/modules/modeling.rst -------------------------------------------------------------------------------- /docs/modules/utils.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/docs/modules/utils.rst -------------------------------------------------------------------------------- /docs/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/docs/requirements.txt -------------------------------------------------------------------------------- /figs/framework.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/figs/framework.png -------------------------------------------------------------------------------- /onnx/.gitignore: -------------------------------------------------------------------------------- 1 | ncnn 2 | -------------------------------------------------------------------------------- /onnx/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/onnx/README.md -------------------------------------------------------------------------------- /onnx/export_model_to_onnx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/onnx/export_model_to_onnx.py -------------------------------------------------------------------------------- /onnx/pytorch-onnx-caffe-ncnn-rt.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/onnx/pytorch-onnx-caffe-ncnn-rt.sh -------------------------------------------------------------------------------- /onnx/pytorch-onnx-caffe-ncnn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/onnx/pytorch-onnx-caffe-ncnn.sh -------------------------------------------------------------------------------- /onnx/test_onnxruntime.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/onnx/test_onnxruntime.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/setup.py -------------------------------------------------------------------------------- /tools/compute_flops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/tools/compute_flops.py -------------------------------------------------------------------------------- /tools/convert_fcos_weight.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/tools/convert_fcos_weight.py -------------------------------------------------------------------------------- /tools/remove_optim_from_ckpt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/tools/remove_optim_from_ckpt.py -------------------------------------------------------------------------------- /tools/rename_blendmask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/tools/rename_blendmask.py -------------------------------------------------------------------------------- /tools/train_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/tools/train_net.py -------------------------------------------------------------------------------- /tools/visualize_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FangShancheng/ABINet-PP/HEAD/tools/visualize_data.py --------------------------------------------------------------------------------