├── README.MD ├── cfg ├── fasternet_l.yaml ├── fasternet_m.yaml ├── fasternet_s.yaml ├── fasternet_t0.yaml ├── fasternet_t1.yaml └── fasternet_t2.yaml ├── data ├── __init__.py ├── custom_imagenet_data.py └── data_api.py ├── detection ├── README.MD ├── backbones │ ├── __init__.py │ └── fasternet.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 │ └── fasternet │ │ ├── mask_rcnn_fasternet_l_fpn_1x_coco.py │ │ ├── mask_rcnn_fasternet_m_fpn_1x_coco.py │ │ └── mask_rcnn_fasternet_s_fpn_1x_coco.py ├── dist_test.sh ├── dist_train.sh ├── get_flops.py ├── test.py └── train.py ├── models ├── __init__.py ├── fasternet.py ├── model_api.py └── registry.py ├── requirements.txt ├── train_test.py └── utils ├── __init__.py ├── fuse_conv_bn.py ├── loss.py └── utils.py /README.MD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JierunChen/FasterNet/HEAD/README.MD -------------------------------------------------------------------------------- /cfg/fasternet_l.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JierunChen/FasterNet/HEAD/cfg/fasternet_l.yaml -------------------------------------------------------------------------------- /cfg/fasternet_m.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JierunChen/FasterNet/HEAD/cfg/fasternet_m.yaml -------------------------------------------------------------------------------- /cfg/fasternet_s.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JierunChen/FasterNet/HEAD/cfg/fasternet_s.yaml -------------------------------------------------------------------------------- /cfg/fasternet_t0.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JierunChen/FasterNet/HEAD/cfg/fasternet_t0.yaml -------------------------------------------------------------------------------- /cfg/fasternet_t1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JierunChen/FasterNet/HEAD/cfg/fasternet_t1.yaml -------------------------------------------------------------------------------- /cfg/fasternet_t2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JierunChen/FasterNet/HEAD/cfg/fasternet_t2.yaml -------------------------------------------------------------------------------- /data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /data/custom_imagenet_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JierunChen/FasterNet/HEAD/data/custom_imagenet_data.py -------------------------------------------------------------------------------- /data/data_api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JierunChen/FasterNet/HEAD/data/data_api.py -------------------------------------------------------------------------------- /detection/README.MD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JierunChen/FasterNet/HEAD/detection/README.MD -------------------------------------------------------------------------------- /detection/backbones/__init__.py: -------------------------------------------------------------------------------- 1 | from .fasternet import * -------------------------------------------------------------------------------- /detection/backbones/fasternet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JierunChen/FasterNet/HEAD/detection/backbones/fasternet.py -------------------------------------------------------------------------------- /detection/benchmark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JierunChen/FasterNet/HEAD/detection/benchmark.py -------------------------------------------------------------------------------- /detection/configs/_base_/datasets/coco_detection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JierunChen/FasterNet/HEAD/detection/configs/_base_/datasets/coco_detection.py -------------------------------------------------------------------------------- /detection/configs/_base_/datasets/coco_instance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JierunChen/FasterNet/HEAD/detection/configs/_base_/datasets/coco_instance.py -------------------------------------------------------------------------------- /detection/configs/_base_/default_runtime.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JierunChen/FasterNet/HEAD/detection/configs/_base_/default_runtime.py -------------------------------------------------------------------------------- /detection/configs/_base_/models/mask_rcnn_r50_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JierunChen/FasterNet/HEAD/detection/configs/_base_/models/mask_rcnn_r50_fpn.py -------------------------------------------------------------------------------- /detection/configs/_base_/schedules/schedule_1x.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JierunChen/FasterNet/HEAD/detection/configs/_base_/schedules/schedule_1x.py -------------------------------------------------------------------------------- /detection/configs/_base_/schedules/schedule_20e.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JierunChen/FasterNet/HEAD/detection/configs/_base_/schedules/schedule_20e.py -------------------------------------------------------------------------------- /detection/configs/_base_/schedules/schedule_2x.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JierunChen/FasterNet/HEAD/detection/configs/_base_/schedules/schedule_2x.py -------------------------------------------------------------------------------- /detection/configs/fasternet/mask_rcnn_fasternet_l_fpn_1x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JierunChen/FasterNet/HEAD/detection/configs/fasternet/mask_rcnn_fasternet_l_fpn_1x_coco.py -------------------------------------------------------------------------------- /detection/configs/fasternet/mask_rcnn_fasternet_m_fpn_1x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JierunChen/FasterNet/HEAD/detection/configs/fasternet/mask_rcnn_fasternet_m_fpn_1x_coco.py -------------------------------------------------------------------------------- /detection/configs/fasternet/mask_rcnn_fasternet_s_fpn_1x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JierunChen/FasterNet/HEAD/detection/configs/fasternet/mask_rcnn_fasternet_s_fpn_1x_coco.py -------------------------------------------------------------------------------- /detection/dist_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JierunChen/FasterNet/HEAD/detection/dist_test.sh -------------------------------------------------------------------------------- /detection/dist_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JierunChen/FasterNet/HEAD/detection/dist_train.sh -------------------------------------------------------------------------------- /detection/get_flops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JierunChen/FasterNet/HEAD/detection/get_flops.py -------------------------------------------------------------------------------- /detection/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JierunChen/FasterNet/HEAD/detection/test.py -------------------------------------------------------------------------------- /detection/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JierunChen/FasterNet/HEAD/detection/train.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | from .registry import * -------------------------------------------------------------------------------- /models/fasternet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JierunChen/FasterNet/HEAD/models/fasternet.py -------------------------------------------------------------------------------- /models/model_api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JierunChen/FasterNet/HEAD/models/model_api.py -------------------------------------------------------------------------------- /models/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JierunChen/FasterNet/HEAD/models/registry.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JierunChen/FasterNet/HEAD/requirements.txt -------------------------------------------------------------------------------- /train_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JierunChen/FasterNet/HEAD/train_test.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/fuse_conv_bn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JierunChen/FasterNet/HEAD/utils/fuse_conv_bn.py -------------------------------------------------------------------------------- /utils/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JierunChen/FasterNet/HEAD/utils/loss.py -------------------------------------------------------------------------------- /utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JierunChen/FasterNet/HEAD/utils/utils.py --------------------------------------------------------------------------------