├── .gitignore ├── LICENSE ├── README.md ├── data ├── __init__.py ├── datasets.py ├── samplers.py └── threeaugment.py ├── demo ├── feature_map_visualization.ipynb ├── figures │ ├── cat-stage0-block1-conv_i.png │ ├── cat-stage0-block1-conv_l.png │ ├── cat-stage0-block1-conv_m.png │ ├── cat-stage0-block1-conv_s.png │ ├── dog2-stage0-block1-conv_i.png │ ├── dog2-stage0-block1-conv_l.png │ ├── dog2-stage0-block1-conv_m.png │ └── dog2-stage0-block1-conv_s.png └── fused_model_evaluation.ipynb ├── detection ├── .gitignore ├── README.md ├── checkpoint.py ├── configs │ ├── _base_ │ │ ├── datasets │ │ │ ├── cityscapes_detection.py │ │ │ ├── cityscapes_instance.py │ │ │ ├── coco_detection.py │ │ │ ├── coco_instance.py │ │ │ ├── coco_instance_semantic.py │ │ │ ├── deepfashion.py │ │ │ ├── lvis_v0.5_instance.py │ │ │ ├── lvis_v1_instance.py │ │ │ ├── voc0712.py │ │ │ └── wider_face.py │ │ ├── default_runtime.py │ │ ├── models │ │ │ ├── cascade_mask_rcnn_pvtv2_b2_fpn.py │ │ │ ├── cascade_mask_rcnn_r50_fpn.py │ │ │ ├── cascade_rcnn_r50_fpn.py │ │ │ ├── fast_rcnn_r50_fpn.py │ │ │ ├── faster_rcnn_r50_caffe_c4.py │ │ │ ├── faster_rcnn_r50_caffe_dc5.py │ │ │ ├── faster_rcnn_r50_fpn.py │ │ │ ├── mask_rcnn_r50_caffe_c4.py │ │ │ ├── mask_rcnn_r50_fpn.py │ │ │ ├── retinanet_r50_fpn.py │ │ │ ├── rpn_r50_caffe_c4.py │ │ │ ├── rpn_r50_fpn.py │ │ │ └── ssd300.py │ │ └── schedules │ │ │ ├── schedule_1x.py │ │ │ ├── schedule_20e.py │ │ │ └── schedule_2x.py │ ├── mask_rcnn_repnext_m3_fpn_1x_coco.py │ ├── mask_rcnn_repnext_m4_fpn_1x_coco.py │ └── mask_rcnn_repnext_m5_fpn_1x_coco.py ├── dist_test.sh ├── dist_train.sh ├── eval.sh ├── logs │ ├── repnext_m3_coco.json │ ├── repnext_m3_coco_4073.json │ ├── repnext_m4_coco.json │ ├── repnext_m4_coco_4266.json │ └── repnext_m5_coco.json ├── mmcv_custom │ └── runner │ │ ├── checkpoint.py │ │ ├── epoch_based_runner.py │ │ └── optimizer.py ├── mmdet_custom │ └── apis │ │ └── train.py ├── repnext.py ├── slurm_train.sh ├── test.py ├── train.py └── train.sh ├── dist_train_cifar.sh ├── dist_train_sleep.sh ├── engine.py ├── eval.sh ├── export_coreml.py ├── figures ├── ablation_convnext.png ├── architecture.key ├── architecture.png ├── compact_model_experiments.png ├── compact_models_cifar_100.png ├── compact_models_imagenet_1k.png ├── equivalent.png ├── latency.png ├── latency │ ├── repnext_m1_latency.png │ ├── repnext_m2_latency.png │ ├── repnext_m3_latency.png │ ├── repnext_m4_latency.png │ └── repnext_m5_latency.png └── transforms.png ├── flops.py ├── fuse_eval.py ├── logs ├── ablation │ ├── convnext_femto_120e_7237.txt │ ├── convnext_femto_120e_7419.txt │ ├── convnext_femto_120e_7428.txt │ ├── repnext_m1_120e_7534.txt │ └── repnext_m1_120e_7597.txt ├── compact │ ├── cifar │ │ ├── repnext_m0_cifar_100e.txt │ │ ├── repnext_m0_cifar_200e.txt │ │ ├── repnext_m0_cifar_300e.txt │ │ ├── repnext_m0_cifar_400e.txt │ │ ├── repnext_m0_e_cifar_100e.txt │ │ ├── repnext_m0_e_cifar_200e.txt │ │ ├── repnext_m0_e_cifar_300e.txt │ │ ├── repnext_m0_e_cifar_400e.txt │ │ ├── repvit_m0_6_cifar_100e.txt │ │ ├── repvit_m0_6_cifar_200e.txt │ │ ├── repvit_m0_6_cifar_300e.txt │ │ ├── repvit_m0_6_cifar_400e.txt │ │ ├── starnet_s1_cifar_100e.txt │ │ ├── starnet_s1_cifar_200e.txt │ │ ├── starnet_s1_cifar_300e.txt │ │ └── starnet_s1_cifar_400e.txt │ └── imgnet │ │ ├── repnext_m0_e_imgnet_100e.txt │ │ ├── repnext_m0_e_imgnet_200e.txt │ │ ├── repnext_m0_e_imgnet_300e.txt │ │ ├── repnext_m0_e_imgnet_400e.txt │ │ ├── repnext_m0_imgnet_100e.txt │ │ ├── repnext_m0_imgnet_200e.txt │ │ ├── repnext_m0_imgnet_300e.txt │ │ ├── repnext_m0_imgnet_400e.txt │ │ ├── repvit_m0_6_imgnet_100e.txt │ │ ├── repvit_m0_6_imgnet_200e.txt │ │ ├── repvit_m0_6_imgnet_300e.txt │ │ ├── repvit_m0_6_imgnet_400e.txt │ │ ├── starnet_s1_imgnet_100e.txt │ │ ├── starnet_s1_imgnet_200e.txt │ │ ├── starnet_s1_imgnet_300e.txt │ │ └── starnet_s1_imgnet_400e.txt ├── repnext_m0_distill_300e.txt ├── repnext_m0_without_distill_300e.txt ├── repnext_m1_distill_300e.txt ├── repnext_m1_without_distill_300e.txt ├── repnext_m2_distill_300e.txt ├── repnext_m2_without_distill_300e.txt ├── repnext_m3_distill_300e.txt ├── repnext_m3_without_distill_300e.txt ├── repnext_m4_distill_300e.txt ├── repnext_m4_without_distill_300e.txt ├── repnext_m5_distill_300e.txt ├── repnext_m5_without_distill_300e.txt └── strategy │ ├── repnext_m0_sz224_4xbs256_ep300_distill_args.yaml │ ├── repnext_m0_sz224_4xbs256_ep300_distill_summary.csv │ ├── repnext_m0_sz224_4xbs512_ep300_args.yaml │ ├── repnext_m0_sz224_4xbs512_ep300_summary.csv │ ├── repnext_m0e_sz224_4xbs256_ep300_distill_summary.csv │ ├── repnext_m0e_sz224_4xbs512_ep300_summary.csv │ ├── repnext_m1_sz224_4xbs256_ep300_distill_args.yaml │ ├── repnext_m1_sz224_4xbs256_ep300_distill_summary.csv │ ├── repnext_m1_sz224_4xbs512_ep300_args.yaml │ ├── repnext_m1_sz224_4xbs512_ep300_summary.csv │ ├── repnext_m1e_sz224_4xbs256_ep300_distill_summary.csv │ ├── repnext_m1e_sz224_4xbs512_ep300_summary.csv │ ├── repnext_m2_sz224_4xbs256_ep300_distill_args.yaml │ ├── repnext_m2_sz224_4xbs256_ep300_distill_summary.csv │ ├── repnext_m2_sz224_4xbs512_ep300_args.yaml │ ├── repnext_m2_sz224_4xbs512_ep300_summary.csv │ ├── repnext_m2e_sz224_4xbs256_ep300_distill_summary.csv │ └── repnext_m2e_sz224_4xbs512_ep300_summary.csv ├── losses.py ├── main.py ├── model ├── __init__.py ├── repnext.py └── repnext_test.py ├── moganet_dist_train_imgnet.sh ├── moganet_train.py ├── moganet_valid.py ├── requirements.txt ├── segmentation ├── .gitignore ├── README.md ├── align_resize.py ├── configs │ ├── _base_ │ │ ├── datasets │ │ │ └── ade20k.py │ │ ├── default_runtime.py │ │ ├── models │ │ │ └── fpn_r50.py │ │ └── schedules │ │ │ ├── schedule_160k.py │ │ │ ├── schedule_20k.py │ │ │ ├── schedule_40k.py │ │ │ └── schedule_80k.py │ └── sem_fpn │ │ ├── fpn_repnext_m3_ade20k_40k.py │ │ ├── fpn_repnext_m4_ade20k_40k.py │ │ └── fpn_repnext_m5_ade20k_40k.py ├── eval.sh ├── logs │ ├── repnext_m3_ade20k.json │ ├── repnext_m4_ade20k.json │ └── repnext_m5_ade20k.json ├── repnext.py ├── tools │ ├── analyze_logs.py │ ├── benchmark.py │ ├── browse_dataset.py │ ├── convert_datasets │ │ ├── chase_db1.py │ │ ├── cityscapes.py │ │ ├── coco_stuff10k.py │ │ ├── coco_stuff164k.py │ │ ├── drive.py │ │ ├── hrf.py │ │ ├── pascal_context.py │ │ ├── stare.py │ │ └── voc_aug.py │ ├── deploy_test.py │ ├── dist_test.sh │ ├── dist_train.sh │ ├── get_flops.py │ ├── model_converters │ │ ├── mit2mmseg.py │ │ ├── swin2mmseg.py │ │ └── vit2mmseg.py │ ├── onnx2tensorrt.py │ ├── print_config.py │ ├── publish_model.py │ ├── pytorch2onnx.py │ ├── pytorch2torchscript.py │ ├── slurm_test.sh │ ├── slurm_train.sh │ ├── test.py │ ├── torchserve │ │ ├── mmseg2torchserve.py │ │ ├── mmseg_handler.py │ │ └── test_torchserve.py │ └── train.py └── train.sh ├── speed_gpu.py ├── tools ├── benchmark.py ├── compact.py └── latency.py ├── train.sh └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/README.md -------------------------------------------------------------------------------- /data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /data/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/data/datasets.py -------------------------------------------------------------------------------- /data/samplers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/data/samplers.py -------------------------------------------------------------------------------- /data/threeaugment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/data/threeaugment.py -------------------------------------------------------------------------------- /demo/feature_map_visualization.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/demo/feature_map_visualization.ipynb -------------------------------------------------------------------------------- /demo/figures/cat-stage0-block1-conv_i.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/demo/figures/cat-stage0-block1-conv_i.png -------------------------------------------------------------------------------- /demo/figures/cat-stage0-block1-conv_l.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/demo/figures/cat-stage0-block1-conv_l.png -------------------------------------------------------------------------------- /demo/figures/cat-stage0-block1-conv_m.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/demo/figures/cat-stage0-block1-conv_m.png -------------------------------------------------------------------------------- /demo/figures/cat-stage0-block1-conv_s.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/demo/figures/cat-stage0-block1-conv_s.png -------------------------------------------------------------------------------- /demo/figures/dog2-stage0-block1-conv_i.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/demo/figures/dog2-stage0-block1-conv_i.png -------------------------------------------------------------------------------- /demo/figures/dog2-stage0-block1-conv_l.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/demo/figures/dog2-stage0-block1-conv_l.png -------------------------------------------------------------------------------- /demo/figures/dog2-stage0-block1-conv_m.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/demo/figures/dog2-stage0-block1-conv_m.png -------------------------------------------------------------------------------- /demo/figures/dog2-stage0-block1-conv_s.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/demo/figures/dog2-stage0-block1-conv_s.png -------------------------------------------------------------------------------- /demo/fused_model_evaluation.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/demo/fused_model_evaluation.ipynb -------------------------------------------------------------------------------- /detection/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/detection/.gitignore -------------------------------------------------------------------------------- /detection/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/detection/README.md -------------------------------------------------------------------------------- /detection/checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/detection/checkpoint.py -------------------------------------------------------------------------------- /detection/configs/_base_/datasets/cityscapes_detection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/detection/configs/_base_/datasets/cityscapes_detection.py -------------------------------------------------------------------------------- /detection/configs/_base_/datasets/cityscapes_instance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/detection/configs/_base_/datasets/cityscapes_instance.py -------------------------------------------------------------------------------- /detection/configs/_base_/datasets/coco_detection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/detection/configs/_base_/datasets/coco_detection.py -------------------------------------------------------------------------------- /detection/configs/_base_/datasets/coco_instance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/detection/configs/_base_/datasets/coco_instance.py -------------------------------------------------------------------------------- /detection/configs/_base_/datasets/coco_instance_semantic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/detection/configs/_base_/datasets/coco_instance_semantic.py -------------------------------------------------------------------------------- /detection/configs/_base_/datasets/deepfashion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/detection/configs/_base_/datasets/deepfashion.py -------------------------------------------------------------------------------- /detection/configs/_base_/datasets/lvis_v0.5_instance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/detection/configs/_base_/datasets/lvis_v0.5_instance.py -------------------------------------------------------------------------------- /detection/configs/_base_/datasets/lvis_v1_instance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/detection/configs/_base_/datasets/lvis_v1_instance.py -------------------------------------------------------------------------------- /detection/configs/_base_/datasets/voc0712.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/detection/configs/_base_/datasets/voc0712.py -------------------------------------------------------------------------------- /detection/configs/_base_/datasets/wider_face.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/detection/configs/_base_/datasets/wider_face.py -------------------------------------------------------------------------------- /detection/configs/_base_/default_runtime.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/detection/configs/_base_/default_runtime.py -------------------------------------------------------------------------------- /detection/configs/_base_/models/cascade_mask_rcnn_pvtv2_b2_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/detection/configs/_base_/models/cascade_mask_rcnn_pvtv2_b2_fpn.py -------------------------------------------------------------------------------- /detection/configs/_base_/models/cascade_mask_rcnn_r50_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/detection/configs/_base_/models/cascade_mask_rcnn_r50_fpn.py -------------------------------------------------------------------------------- /detection/configs/_base_/models/cascade_rcnn_r50_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/detection/configs/_base_/models/cascade_rcnn_r50_fpn.py -------------------------------------------------------------------------------- /detection/configs/_base_/models/fast_rcnn_r50_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/detection/configs/_base_/models/fast_rcnn_r50_fpn.py -------------------------------------------------------------------------------- /detection/configs/_base_/models/faster_rcnn_r50_caffe_c4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/detection/configs/_base_/models/faster_rcnn_r50_caffe_c4.py -------------------------------------------------------------------------------- /detection/configs/_base_/models/faster_rcnn_r50_caffe_dc5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/detection/configs/_base_/models/faster_rcnn_r50_caffe_dc5.py -------------------------------------------------------------------------------- /detection/configs/_base_/models/faster_rcnn_r50_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/detection/configs/_base_/models/faster_rcnn_r50_fpn.py -------------------------------------------------------------------------------- /detection/configs/_base_/models/mask_rcnn_r50_caffe_c4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/detection/configs/_base_/models/mask_rcnn_r50_caffe_c4.py -------------------------------------------------------------------------------- /detection/configs/_base_/models/mask_rcnn_r50_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/detection/configs/_base_/models/mask_rcnn_r50_fpn.py -------------------------------------------------------------------------------- /detection/configs/_base_/models/retinanet_r50_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/detection/configs/_base_/models/retinanet_r50_fpn.py -------------------------------------------------------------------------------- /detection/configs/_base_/models/rpn_r50_caffe_c4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/detection/configs/_base_/models/rpn_r50_caffe_c4.py -------------------------------------------------------------------------------- /detection/configs/_base_/models/rpn_r50_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/detection/configs/_base_/models/rpn_r50_fpn.py -------------------------------------------------------------------------------- /detection/configs/_base_/models/ssd300.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/detection/configs/_base_/models/ssd300.py -------------------------------------------------------------------------------- /detection/configs/_base_/schedules/schedule_1x.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/detection/configs/_base_/schedules/schedule_1x.py -------------------------------------------------------------------------------- /detection/configs/_base_/schedules/schedule_20e.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/detection/configs/_base_/schedules/schedule_20e.py -------------------------------------------------------------------------------- /detection/configs/_base_/schedules/schedule_2x.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/detection/configs/_base_/schedules/schedule_2x.py -------------------------------------------------------------------------------- /detection/configs/mask_rcnn_repnext_m3_fpn_1x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/detection/configs/mask_rcnn_repnext_m3_fpn_1x_coco.py -------------------------------------------------------------------------------- /detection/configs/mask_rcnn_repnext_m4_fpn_1x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/detection/configs/mask_rcnn_repnext_m4_fpn_1x_coco.py -------------------------------------------------------------------------------- /detection/configs/mask_rcnn_repnext_m5_fpn_1x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/detection/configs/mask_rcnn_repnext_m5_fpn_1x_coco.py -------------------------------------------------------------------------------- /detection/dist_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/detection/dist_test.sh -------------------------------------------------------------------------------- /detection/dist_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/detection/dist_train.sh -------------------------------------------------------------------------------- /detection/eval.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/detection/eval.sh -------------------------------------------------------------------------------- /detection/logs/repnext_m3_coco.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/detection/logs/repnext_m3_coco.json -------------------------------------------------------------------------------- /detection/logs/repnext_m3_coco_4073.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/detection/logs/repnext_m3_coco_4073.json -------------------------------------------------------------------------------- /detection/logs/repnext_m4_coco.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/detection/logs/repnext_m4_coco.json -------------------------------------------------------------------------------- /detection/logs/repnext_m4_coco_4266.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/detection/logs/repnext_m4_coco_4266.json -------------------------------------------------------------------------------- /detection/logs/repnext_m5_coco.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/detection/logs/repnext_m5_coco.json -------------------------------------------------------------------------------- /detection/mmcv_custom/runner/checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/detection/mmcv_custom/runner/checkpoint.py -------------------------------------------------------------------------------- /detection/mmcv_custom/runner/epoch_based_runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/detection/mmcv_custom/runner/epoch_based_runner.py -------------------------------------------------------------------------------- /detection/mmcv_custom/runner/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/detection/mmcv_custom/runner/optimizer.py -------------------------------------------------------------------------------- /detection/mmdet_custom/apis/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/detection/mmdet_custom/apis/train.py -------------------------------------------------------------------------------- /detection/repnext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/detection/repnext.py -------------------------------------------------------------------------------- /detection/slurm_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/detection/slurm_train.sh -------------------------------------------------------------------------------- /detection/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/detection/test.py -------------------------------------------------------------------------------- /detection/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/detection/train.py -------------------------------------------------------------------------------- /detection/train.sh: -------------------------------------------------------------------------------- 1 | ./dist_train.sh configs/mask_rcnn_repnext_m3_fpn_1x_coco.py 8 -------------------------------------------------------------------------------- /dist_train_cifar.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/dist_train_cifar.sh -------------------------------------------------------------------------------- /dist_train_sleep.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/dist_train_sleep.sh -------------------------------------------------------------------------------- /engine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/engine.py -------------------------------------------------------------------------------- /eval.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/eval.sh -------------------------------------------------------------------------------- /export_coreml.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/export_coreml.py -------------------------------------------------------------------------------- /figures/ablation_convnext.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/figures/ablation_convnext.png -------------------------------------------------------------------------------- /figures/architecture.key: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/figures/architecture.key -------------------------------------------------------------------------------- /figures/architecture.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/figures/architecture.png -------------------------------------------------------------------------------- /figures/compact_model_experiments.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/figures/compact_model_experiments.png -------------------------------------------------------------------------------- /figures/compact_models_cifar_100.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/figures/compact_models_cifar_100.png -------------------------------------------------------------------------------- /figures/compact_models_imagenet_1k.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/figures/compact_models_imagenet_1k.png -------------------------------------------------------------------------------- /figures/equivalent.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/figures/equivalent.png -------------------------------------------------------------------------------- /figures/latency.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/figures/latency.png -------------------------------------------------------------------------------- /figures/latency/repnext_m1_latency.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/figures/latency/repnext_m1_latency.png -------------------------------------------------------------------------------- /figures/latency/repnext_m2_latency.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/figures/latency/repnext_m2_latency.png -------------------------------------------------------------------------------- /figures/latency/repnext_m3_latency.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/figures/latency/repnext_m3_latency.png -------------------------------------------------------------------------------- /figures/latency/repnext_m4_latency.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/figures/latency/repnext_m4_latency.png -------------------------------------------------------------------------------- /figures/latency/repnext_m5_latency.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/figures/latency/repnext_m5_latency.png -------------------------------------------------------------------------------- /figures/transforms.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/figures/transforms.png -------------------------------------------------------------------------------- /flops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/flops.py -------------------------------------------------------------------------------- /fuse_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/fuse_eval.py -------------------------------------------------------------------------------- /logs/ablation/convnext_femto_120e_7237.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/logs/ablation/convnext_femto_120e_7237.txt -------------------------------------------------------------------------------- /logs/ablation/convnext_femto_120e_7419.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/logs/ablation/convnext_femto_120e_7419.txt -------------------------------------------------------------------------------- /logs/ablation/convnext_femto_120e_7428.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/logs/ablation/convnext_femto_120e_7428.txt -------------------------------------------------------------------------------- /logs/ablation/repnext_m1_120e_7534.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/logs/ablation/repnext_m1_120e_7534.txt -------------------------------------------------------------------------------- /logs/ablation/repnext_m1_120e_7597.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/logs/ablation/repnext_m1_120e_7597.txt -------------------------------------------------------------------------------- /logs/compact/cifar/repnext_m0_cifar_100e.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/logs/compact/cifar/repnext_m0_cifar_100e.txt -------------------------------------------------------------------------------- /logs/compact/cifar/repnext_m0_cifar_200e.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/logs/compact/cifar/repnext_m0_cifar_200e.txt -------------------------------------------------------------------------------- /logs/compact/cifar/repnext_m0_cifar_300e.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/logs/compact/cifar/repnext_m0_cifar_300e.txt -------------------------------------------------------------------------------- /logs/compact/cifar/repnext_m0_cifar_400e.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/logs/compact/cifar/repnext_m0_cifar_400e.txt -------------------------------------------------------------------------------- /logs/compact/cifar/repnext_m0_e_cifar_100e.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/logs/compact/cifar/repnext_m0_e_cifar_100e.txt -------------------------------------------------------------------------------- /logs/compact/cifar/repnext_m0_e_cifar_200e.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/logs/compact/cifar/repnext_m0_e_cifar_200e.txt -------------------------------------------------------------------------------- /logs/compact/cifar/repnext_m0_e_cifar_300e.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/logs/compact/cifar/repnext_m0_e_cifar_300e.txt -------------------------------------------------------------------------------- /logs/compact/cifar/repnext_m0_e_cifar_400e.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/logs/compact/cifar/repnext_m0_e_cifar_400e.txt -------------------------------------------------------------------------------- /logs/compact/cifar/repvit_m0_6_cifar_100e.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/logs/compact/cifar/repvit_m0_6_cifar_100e.txt -------------------------------------------------------------------------------- /logs/compact/cifar/repvit_m0_6_cifar_200e.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/logs/compact/cifar/repvit_m0_6_cifar_200e.txt -------------------------------------------------------------------------------- /logs/compact/cifar/repvit_m0_6_cifar_300e.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/logs/compact/cifar/repvit_m0_6_cifar_300e.txt -------------------------------------------------------------------------------- /logs/compact/cifar/repvit_m0_6_cifar_400e.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/logs/compact/cifar/repvit_m0_6_cifar_400e.txt -------------------------------------------------------------------------------- /logs/compact/cifar/starnet_s1_cifar_100e.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/logs/compact/cifar/starnet_s1_cifar_100e.txt -------------------------------------------------------------------------------- /logs/compact/cifar/starnet_s1_cifar_200e.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/logs/compact/cifar/starnet_s1_cifar_200e.txt -------------------------------------------------------------------------------- /logs/compact/cifar/starnet_s1_cifar_300e.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/logs/compact/cifar/starnet_s1_cifar_300e.txt -------------------------------------------------------------------------------- /logs/compact/cifar/starnet_s1_cifar_400e.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/logs/compact/cifar/starnet_s1_cifar_400e.txt -------------------------------------------------------------------------------- /logs/compact/imgnet/repnext_m0_e_imgnet_100e.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/logs/compact/imgnet/repnext_m0_e_imgnet_100e.txt -------------------------------------------------------------------------------- /logs/compact/imgnet/repnext_m0_e_imgnet_200e.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/logs/compact/imgnet/repnext_m0_e_imgnet_200e.txt -------------------------------------------------------------------------------- /logs/compact/imgnet/repnext_m0_e_imgnet_300e.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/logs/compact/imgnet/repnext_m0_e_imgnet_300e.txt -------------------------------------------------------------------------------- /logs/compact/imgnet/repnext_m0_e_imgnet_400e.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/logs/compact/imgnet/repnext_m0_e_imgnet_400e.txt -------------------------------------------------------------------------------- /logs/compact/imgnet/repnext_m0_imgnet_100e.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/logs/compact/imgnet/repnext_m0_imgnet_100e.txt -------------------------------------------------------------------------------- /logs/compact/imgnet/repnext_m0_imgnet_200e.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/logs/compact/imgnet/repnext_m0_imgnet_200e.txt -------------------------------------------------------------------------------- /logs/compact/imgnet/repnext_m0_imgnet_300e.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/logs/compact/imgnet/repnext_m0_imgnet_300e.txt -------------------------------------------------------------------------------- /logs/compact/imgnet/repnext_m0_imgnet_400e.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/logs/compact/imgnet/repnext_m0_imgnet_400e.txt -------------------------------------------------------------------------------- /logs/compact/imgnet/repvit_m0_6_imgnet_100e.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/logs/compact/imgnet/repvit_m0_6_imgnet_100e.txt -------------------------------------------------------------------------------- /logs/compact/imgnet/repvit_m0_6_imgnet_200e.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/logs/compact/imgnet/repvit_m0_6_imgnet_200e.txt -------------------------------------------------------------------------------- /logs/compact/imgnet/repvit_m0_6_imgnet_300e.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/logs/compact/imgnet/repvit_m0_6_imgnet_300e.txt -------------------------------------------------------------------------------- /logs/compact/imgnet/repvit_m0_6_imgnet_400e.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/logs/compact/imgnet/repvit_m0_6_imgnet_400e.txt -------------------------------------------------------------------------------- /logs/compact/imgnet/starnet_s1_imgnet_100e.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/logs/compact/imgnet/starnet_s1_imgnet_100e.txt -------------------------------------------------------------------------------- /logs/compact/imgnet/starnet_s1_imgnet_200e.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/logs/compact/imgnet/starnet_s1_imgnet_200e.txt -------------------------------------------------------------------------------- /logs/compact/imgnet/starnet_s1_imgnet_300e.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/logs/compact/imgnet/starnet_s1_imgnet_300e.txt -------------------------------------------------------------------------------- /logs/compact/imgnet/starnet_s1_imgnet_400e.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/logs/compact/imgnet/starnet_s1_imgnet_400e.txt -------------------------------------------------------------------------------- /logs/repnext_m0_distill_300e.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/logs/repnext_m0_distill_300e.txt -------------------------------------------------------------------------------- /logs/repnext_m0_without_distill_300e.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/logs/repnext_m0_without_distill_300e.txt -------------------------------------------------------------------------------- /logs/repnext_m1_distill_300e.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/logs/repnext_m1_distill_300e.txt -------------------------------------------------------------------------------- /logs/repnext_m1_without_distill_300e.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/logs/repnext_m1_without_distill_300e.txt -------------------------------------------------------------------------------- /logs/repnext_m2_distill_300e.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/logs/repnext_m2_distill_300e.txt -------------------------------------------------------------------------------- /logs/repnext_m2_without_distill_300e.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/logs/repnext_m2_without_distill_300e.txt -------------------------------------------------------------------------------- /logs/repnext_m3_distill_300e.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/logs/repnext_m3_distill_300e.txt -------------------------------------------------------------------------------- /logs/repnext_m3_without_distill_300e.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/logs/repnext_m3_without_distill_300e.txt -------------------------------------------------------------------------------- /logs/repnext_m4_distill_300e.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/logs/repnext_m4_distill_300e.txt -------------------------------------------------------------------------------- /logs/repnext_m4_without_distill_300e.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/logs/repnext_m4_without_distill_300e.txt -------------------------------------------------------------------------------- /logs/repnext_m5_distill_300e.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/logs/repnext_m5_distill_300e.txt -------------------------------------------------------------------------------- /logs/repnext_m5_without_distill_300e.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/logs/repnext_m5_without_distill_300e.txt -------------------------------------------------------------------------------- /logs/strategy/repnext_m0_sz224_4xbs256_ep300_distill_args.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/logs/strategy/repnext_m0_sz224_4xbs256_ep300_distill_args.yaml -------------------------------------------------------------------------------- /logs/strategy/repnext_m0_sz224_4xbs256_ep300_distill_summary.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/logs/strategy/repnext_m0_sz224_4xbs256_ep300_distill_summary.csv -------------------------------------------------------------------------------- /logs/strategy/repnext_m0_sz224_4xbs512_ep300_args.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/logs/strategy/repnext_m0_sz224_4xbs512_ep300_args.yaml -------------------------------------------------------------------------------- /logs/strategy/repnext_m0_sz224_4xbs512_ep300_summary.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/logs/strategy/repnext_m0_sz224_4xbs512_ep300_summary.csv -------------------------------------------------------------------------------- /logs/strategy/repnext_m0e_sz224_4xbs256_ep300_distill_summary.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/logs/strategy/repnext_m0e_sz224_4xbs256_ep300_distill_summary.csv -------------------------------------------------------------------------------- /logs/strategy/repnext_m0e_sz224_4xbs512_ep300_summary.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/logs/strategy/repnext_m0e_sz224_4xbs512_ep300_summary.csv -------------------------------------------------------------------------------- /logs/strategy/repnext_m1_sz224_4xbs256_ep300_distill_args.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/logs/strategy/repnext_m1_sz224_4xbs256_ep300_distill_args.yaml -------------------------------------------------------------------------------- /logs/strategy/repnext_m1_sz224_4xbs256_ep300_distill_summary.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/logs/strategy/repnext_m1_sz224_4xbs256_ep300_distill_summary.csv -------------------------------------------------------------------------------- /logs/strategy/repnext_m1_sz224_4xbs512_ep300_args.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/logs/strategy/repnext_m1_sz224_4xbs512_ep300_args.yaml -------------------------------------------------------------------------------- /logs/strategy/repnext_m1_sz224_4xbs512_ep300_summary.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/logs/strategy/repnext_m1_sz224_4xbs512_ep300_summary.csv -------------------------------------------------------------------------------- /logs/strategy/repnext_m1e_sz224_4xbs256_ep300_distill_summary.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/logs/strategy/repnext_m1e_sz224_4xbs256_ep300_distill_summary.csv -------------------------------------------------------------------------------- /logs/strategy/repnext_m1e_sz224_4xbs512_ep300_summary.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/logs/strategy/repnext_m1e_sz224_4xbs512_ep300_summary.csv -------------------------------------------------------------------------------- /logs/strategy/repnext_m2_sz224_4xbs256_ep300_distill_args.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/logs/strategy/repnext_m2_sz224_4xbs256_ep300_distill_args.yaml -------------------------------------------------------------------------------- /logs/strategy/repnext_m2_sz224_4xbs256_ep300_distill_summary.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/logs/strategy/repnext_m2_sz224_4xbs256_ep300_distill_summary.csv -------------------------------------------------------------------------------- /logs/strategy/repnext_m2_sz224_4xbs512_ep300_args.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/logs/strategy/repnext_m2_sz224_4xbs512_ep300_args.yaml -------------------------------------------------------------------------------- /logs/strategy/repnext_m2_sz224_4xbs512_ep300_summary.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/logs/strategy/repnext_m2_sz224_4xbs512_ep300_summary.csv -------------------------------------------------------------------------------- /logs/strategy/repnext_m2e_sz224_4xbs256_ep300_distill_summary.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/logs/strategy/repnext_m2e_sz224_4xbs256_ep300_distill_summary.csv -------------------------------------------------------------------------------- /logs/strategy/repnext_m2e_sz224_4xbs512_ep300_summary.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/logs/strategy/repnext_m2e_sz224_4xbs512_ep300_summary.csv -------------------------------------------------------------------------------- /losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/losses.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/main.py -------------------------------------------------------------------------------- /model/__init__.py: -------------------------------------------------------------------------------- 1 | import model.repnext 2 | -------------------------------------------------------------------------------- /model/repnext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/model/repnext.py -------------------------------------------------------------------------------- /model/repnext_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/model/repnext_test.py -------------------------------------------------------------------------------- /moganet_dist_train_imgnet.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/moganet_dist_train_imgnet.sh -------------------------------------------------------------------------------- /moganet_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/moganet_train.py -------------------------------------------------------------------------------- /moganet_valid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/moganet_valid.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | torch 2 | timm==0.5.4 3 | fvcore -------------------------------------------------------------------------------- /segmentation/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/segmentation/.gitignore -------------------------------------------------------------------------------- /segmentation/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/segmentation/README.md -------------------------------------------------------------------------------- /segmentation/align_resize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/segmentation/align_resize.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/datasets/ade20k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/segmentation/configs/_base_/datasets/ade20k.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/default_runtime.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/segmentation/configs/_base_/default_runtime.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/models/fpn_r50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/segmentation/configs/_base_/models/fpn_r50.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/schedules/schedule_160k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/segmentation/configs/_base_/schedules/schedule_160k.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/schedules/schedule_20k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/segmentation/configs/_base_/schedules/schedule_20k.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/schedules/schedule_40k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/segmentation/configs/_base_/schedules/schedule_40k.py -------------------------------------------------------------------------------- /segmentation/configs/_base_/schedules/schedule_80k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/segmentation/configs/_base_/schedules/schedule_80k.py -------------------------------------------------------------------------------- /segmentation/configs/sem_fpn/fpn_repnext_m3_ade20k_40k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/segmentation/configs/sem_fpn/fpn_repnext_m3_ade20k_40k.py -------------------------------------------------------------------------------- /segmentation/configs/sem_fpn/fpn_repnext_m4_ade20k_40k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/segmentation/configs/sem_fpn/fpn_repnext_m4_ade20k_40k.py -------------------------------------------------------------------------------- /segmentation/configs/sem_fpn/fpn_repnext_m5_ade20k_40k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/segmentation/configs/sem_fpn/fpn_repnext_m5_ade20k_40k.py -------------------------------------------------------------------------------- /segmentation/eval.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/segmentation/eval.sh -------------------------------------------------------------------------------- /segmentation/logs/repnext_m3_ade20k.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/segmentation/logs/repnext_m3_ade20k.json -------------------------------------------------------------------------------- /segmentation/logs/repnext_m4_ade20k.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/segmentation/logs/repnext_m4_ade20k.json -------------------------------------------------------------------------------- /segmentation/logs/repnext_m5_ade20k.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/segmentation/logs/repnext_m5_ade20k.json -------------------------------------------------------------------------------- /segmentation/repnext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/segmentation/repnext.py -------------------------------------------------------------------------------- /segmentation/tools/analyze_logs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/segmentation/tools/analyze_logs.py -------------------------------------------------------------------------------- /segmentation/tools/benchmark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/segmentation/tools/benchmark.py -------------------------------------------------------------------------------- /segmentation/tools/browse_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/segmentation/tools/browse_dataset.py -------------------------------------------------------------------------------- /segmentation/tools/convert_datasets/chase_db1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/segmentation/tools/convert_datasets/chase_db1.py -------------------------------------------------------------------------------- /segmentation/tools/convert_datasets/cityscapes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/segmentation/tools/convert_datasets/cityscapes.py -------------------------------------------------------------------------------- /segmentation/tools/convert_datasets/coco_stuff10k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/segmentation/tools/convert_datasets/coco_stuff10k.py -------------------------------------------------------------------------------- /segmentation/tools/convert_datasets/coco_stuff164k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/segmentation/tools/convert_datasets/coco_stuff164k.py -------------------------------------------------------------------------------- /segmentation/tools/convert_datasets/drive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/segmentation/tools/convert_datasets/drive.py -------------------------------------------------------------------------------- /segmentation/tools/convert_datasets/hrf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/segmentation/tools/convert_datasets/hrf.py -------------------------------------------------------------------------------- /segmentation/tools/convert_datasets/pascal_context.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/segmentation/tools/convert_datasets/pascal_context.py -------------------------------------------------------------------------------- /segmentation/tools/convert_datasets/stare.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/segmentation/tools/convert_datasets/stare.py -------------------------------------------------------------------------------- /segmentation/tools/convert_datasets/voc_aug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/segmentation/tools/convert_datasets/voc_aug.py -------------------------------------------------------------------------------- /segmentation/tools/deploy_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/segmentation/tools/deploy_test.py -------------------------------------------------------------------------------- /segmentation/tools/dist_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/segmentation/tools/dist_test.sh -------------------------------------------------------------------------------- /segmentation/tools/dist_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/segmentation/tools/dist_train.sh -------------------------------------------------------------------------------- /segmentation/tools/get_flops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/segmentation/tools/get_flops.py -------------------------------------------------------------------------------- /segmentation/tools/model_converters/mit2mmseg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/segmentation/tools/model_converters/mit2mmseg.py -------------------------------------------------------------------------------- /segmentation/tools/model_converters/swin2mmseg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/segmentation/tools/model_converters/swin2mmseg.py -------------------------------------------------------------------------------- /segmentation/tools/model_converters/vit2mmseg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/segmentation/tools/model_converters/vit2mmseg.py -------------------------------------------------------------------------------- /segmentation/tools/onnx2tensorrt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/segmentation/tools/onnx2tensorrt.py -------------------------------------------------------------------------------- /segmentation/tools/print_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/segmentation/tools/print_config.py -------------------------------------------------------------------------------- /segmentation/tools/publish_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/segmentation/tools/publish_model.py -------------------------------------------------------------------------------- /segmentation/tools/pytorch2onnx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/segmentation/tools/pytorch2onnx.py -------------------------------------------------------------------------------- /segmentation/tools/pytorch2torchscript.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/segmentation/tools/pytorch2torchscript.py -------------------------------------------------------------------------------- /segmentation/tools/slurm_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/segmentation/tools/slurm_test.sh -------------------------------------------------------------------------------- /segmentation/tools/slurm_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/segmentation/tools/slurm_train.sh -------------------------------------------------------------------------------- /segmentation/tools/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/segmentation/tools/test.py -------------------------------------------------------------------------------- /segmentation/tools/torchserve/mmseg2torchserve.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/segmentation/tools/torchserve/mmseg2torchserve.py -------------------------------------------------------------------------------- /segmentation/tools/torchserve/mmseg_handler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/segmentation/tools/torchserve/mmseg_handler.py -------------------------------------------------------------------------------- /segmentation/tools/torchserve/test_torchserve.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/segmentation/tools/torchserve/test_torchserve.py -------------------------------------------------------------------------------- /segmentation/tools/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/segmentation/tools/train.py -------------------------------------------------------------------------------- /segmentation/train.sh: -------------------------------------------------------------------------------- 1 | ./tools/dist_train.sh configs/sem_fpn/fpn_repnext_m3_ade20k_40k.py 8 2 | -------------------------------------------------------------------------------- /speed_gpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/speed_gpu.py -------------------------------------------------------------------------------- /tools/benchmark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/tools/benchmark.py -------------------------------------------------------------------------------- /tools/compact.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/tools/compact.py -------------------------------------------------------------------------------- /tools/latency.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/tools/latency.py -------------------------------------------------------------------------------- /train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/train.sh -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suous/RepNeXt/HEAD/utils.py --------------------------------------------------------------------------------