├── LICENSE ├── README.md ├── cfg ├── PartialNet_l.yaml ├── PartialNet_m.yaml ├── PartialNet_s.yaml ├── PartialNet_t0.yaml ├── PartialNet_t1.yaml ├── PartialNet_t2.yaml └── cifar10-PartialNet_to.yaml ├── data ├── __init__.py ├── custom_imagenet_data.py └── data_api.py ├── detection ├── backbones │ ├── __init__.py │ └── partialnet.py ├── benchmark.py ├── configs │ ├── _base_ │ │ ├── datasets │ │ │ ├── coco_detection.py │ │ │ └── coco_instance.py │ │ ├── default_runtime.py │ │ ├── models │ │ │ └── mask_rcnn_r50_fpn.py │ │ └── schedules │ │ │ ├── schedule_1x.py │ │ │ ├── schedule_20e.py │ │ │ └── schedule_2x.py │ └── partialnet │ │ ├── mask_rcnn_partialnet_l_fpn_1x_coco.py │ │ ├── mask_rcnn_partialnet_m_fpn_1x_coco.py │ │ └── mask_rcnn_partialnet_s_fpn_1x_coco.py ├── get_flops.py ├── test.py └── train.py ├── images ├── acc_throughput.png ├── diff_modul.png └── network_overview.png ├── models ├── __init__.py ├── cond_conv2d.py ├── conv2d_same.py ├── convnext.py ├── create_conv2d.py ├── dgconv.py ├── irpe.py ├── mixed_conv2d.py ├── mlp.py ├── mobilenetv2.py ├── model_api.py ├── model_entrypoints.txt ├── partialnet.py ├── registry.py └── resnet.py ├── requirements.txt ├── train_test.py └── utils ├── __init__.py ├── fuse_conv_bn.py ├── loss.py └── utils.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiduo/PartialNet/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiduo/PartialNet/HEAD/README.md -------------------------------------------------------------------------------- /cfg/PartialNet_l.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiduo/PartialNet/HEAD/cfg/PartialNet_l.yaml -------------------------------------------------------------------------------- /cfg/PartialNet_m.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiduo/PartialNet/HEAD/cfg/PartialNet_m.yaml -------------------------------------------------------------------------------- /cfg/PartialNet_s.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiduo/PartialNet/HEAD/cfg/PartialNet_s.yaml -------------------------------------------------------------------------------- /cfg/PartialNet_t0.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiduo/PartialNet/HEAD/cfg/PartialNet_t0.yaml -------------------------------------------------------------------------------- /cfg/PartialNet_t1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiduo/PartialNet/HEAD/cfg/PartialNet_t1.yaml -------------------------------------------------------------------------------- /cfg/PartialNet_t2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiduo/PartialNet/HEAD/cfg/PartialNet_t2.yaml -------------------------------------------------------------------------------- /cfg/cifar10-PartialNet_to.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiduo/PartialNet/HEAD/cfg/cifar10-PartialNet_to.yaml -------------------------------------------------------------------------------- /data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /data/custom_imagenet_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiduo/PartialNet/HEAD/data/custom_imagenet_data.py -------------------------------------------------------------------------------- /data/data_api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiduo/PartialNet/HEAD/data/data_api.py -------------------------------------------------------------------------------- /detection/backbones/__init__.py: -------------------------------------------------------------------------------- 1 | from .partialnet import * -------------------------------------------------------------------------------- /detection/backbones/partialnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiduo/PartialNet/HEAD/detection/backbones/partialnet.py -------------------------------------------------------------------------------- /detection/benchmark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiduo/PartialNet/HEAD/detection/benchmark.py -------------------------------------------------------------------------------- /detection/configs/_base_/datasets/coco_detection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiduo/PartialNet/HEAD/detection/configs/_base_/datasets/coco_detection.py -------------------------------------------------------------------------------- /detection/configs/_base_/datasets/coco_instance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiduo/PartialNet/HEAD/detection/configs/_base_/datasets/coco_instance.py -------------------------------------------------------------------------------- /detection/configs/_base_/default_runtime.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiduo/PartialNet/HEAD/detection/configs/_base_/default_runtime.py -------------------------------------------------------------------------------- /detection/configs/_base_/models/mask_rcnn_r50_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiduo/PartialNet/HEAD/detection/configs/_base_/models/mask_rcnn_r50_fpn.py -------------------------------------------------------------------------------- /detection/configs/_base_/schedules/schedule_1x.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiduo/PartialNet/HEAD/detection/configs/_base_/schedules/schedule_1x.py -------------------------------------------------------------------------------- /detection/configs/_base_/schedules/schedule_20e.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiduo/PartialNet/HEAD/detection/configs/_base_/schedules/schedule_20e.py -------------------------------------------------------------------------------- /detection/configs/_base_/schedules/schedule_2x.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiduo/PartialNet/HEAD/detection/configs/_base_/schedules/schedule_2x.py -------------------------------------------------------------------------------- /detection/configs/partialnet/mask_rcnn_partialnet_l_fpn_1x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiduo/PartialNet/HEAD/detection/configs/partialnet/mask_rcnn_partialnet_l_fpn_1x_coco.py -------------------------------------------------------------------------------- /detection/configs/partialnet/mask_rcnn_partialnet_m_fpn_1x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiduo/PartialNet/HEAD/detection/configs/partialnet/mask_rcnn_partialnet_m_fpn_1x_coco.py -------------------------------------------------------------------------------- /detection/configs/partialnet/mask_rcnn_partialnet_s_fpn_1x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiduo/PartialNet/HEAD/detection/configs/partialnet/mask_rcnn_partialnet_s_fpn_1x_coco.py -------------------------------------------------------------------------------- /detection/get_flops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiduo/PartialNet/HEAD/detection/get_flops.py -------------------------------------------------------------------------------- /detection/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiduo/PartialNet/HEAD/detection/test.py -------------------------------------------------------------------------------- /detection/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiduo/PartialNet/HEAD/detection/train.py -------------------------------------------------------------------------------- /images/acc_throughput.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiduo/PartialNet/HEAD/images/acc_throughput.png -------------------------------------------------------------------------------- /images/diff_modul.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiduo/PartialNet/HEAD/images/diff_modul.png -------------------------------------------------------------------------------- /images/network_overview.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiduo/PartialNet/HEAD/images/network_overview.png -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | from .registry import * -------------------------------------------------------------------------------- /models/cond_conv2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiduo/PartialNet/HEAD/models/cond_conv2d.py -------------------------------------------------------------------------------- /models/conv2d_same.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiduo/PartialNet/HEAD/models/conv2d_same.py -------------------------------------------------------------------------------- /models/convnext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiduo/PartialNet/HEAD/models/convnext.py -------------------------------------------------------------------------------- /models/create_conv2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiduo/PartialNet/HEAD/models/create_conv2d.py -------------------------------------------------------------------------------- /models/dgconv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiduo/PartialNet/HEAD/models/dgconv.py -------------------------------------------------------------------------------- /models/irpe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiduo/PartialNet/HEAD/models/irpe.py -------------------------------------------------------------------------------- /models/mixed_conv2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiduo/PartialNet/HEAD/models/mixed_conv2d.py -------------------------------------------------------------------------------- /models/mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiduo/PartialNet/HEAD/models/mlp.py -------------------------------------------------------------------------------- /models/mobilenetv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiduo/PartialNet/HEAD/models/mobilenetv2.py -------------------------------------------------------------------------------- /models/model_api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiduo/PartialNet/HEAD/models/model_api.py -------------------------------------------------------------------------------- /models/model_entrypoints.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiduo/PartialNet/HEAD/models/model_entrypoints.txt -------------------------------------------------------------------------------- /models/partialnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiduo/PartialNet/HEAD/models/partialnet.py -------------------------------------------------------------------------------- /models/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiduo/PartialNet/HEAD/models/registry.py -------------------------------------------------------------------------------- /models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiduo/PartialNet/HEAD/models/resnet.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiduo/PartialNet/HEAD/requirements.txt -------------------------------------------------------------------------------- /train_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiduo/PartialNet/HEAD/train_test.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/fuse_conv_bn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiduo/PartialNet/HEAD/utils/fuse_conv_bn.py -------------------------------------------------------------------------------- /utils/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiduo/PartialNet/HEAD/utils/loss.py -------------------------------------------------------------------------------- /utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haiduo/PartialNet/HEAD/utils/utils.py --------------------------------------------------------------------------------