├── .github └── workflows │ └── publish-on-main-commit.yaml ├── .gitignore ├── LICENSE ├── README.md ├── assets ├── figures │ ├── applications.png │ ├── diagram.png │ ├── latency_mem.png │ ├── mcunet_demo.gif │ ├── memory_size.png │ └── overview.png └── sample_images │ └── person_det.jpg ├── eval_det.py ├── eval_tflite.py ├── eval_torch.py ├── mcunet ├── __init__.py ├── model_zoo.py ├── tinyengine │ └── README.md ├── tinynas │ ├── __init__.py │ ├── elastic_nn │ │ ├── __init__.py │ │ ├── modules │ │ │ ├── __init__.py │ │ │ ├── dynamic_layers.py │ │ │ └── dynamic_op.py │ │ └── networks │ │ │ ├── __init__.py │ │ │ └── ofa_proxyless.py │ ├── nn │ │ ├── __init__.py │ │ ├── modules │ │ │ ├── __init__.py │ │ │ └── layers.py │ │ └── networks │ │ │ ├── __init__.py │ │ │ ├── mobilenet_v2.py │ │ │ └── proxyless_nets.py │ └── tf_codebase │ │ ├── __init__.py │ │ ├── generate_tflite.py │ │ ├── tf_layers.py │ │ ├── tf_model_zoo.py │ │ └── tf_modules.py └── utils │ ├── __init__.py │ ├── bn_utils.py │ ├── common_tools.py │ ├── det_helper.py │ ├── my_modules.py │ ├── net_config.py │ ├── pytorch_modules.py │ └── pytorch_utils.py ├── requirements.txt └── setup.py /.github/workflows/publish-on-main-commit.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/mcunet/HEAD/.github/workflows/publish-on-main-commit.yaml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/mcunet/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/mcunet/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/mcunet/HEAD/README.md -------------------------------------------------------------------------------- /assets/figures/applications.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/mcunet/HEAD/assets/figures/applications.png -------------------------------------------------------------------------------- /assets/figures/diagram.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/mcunet/HEAD/assets/figures/diagram.png -------------------------------------------------------------------------------- /assets/figures/latency_mem.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/mcunet/HEAD/assets/figures/latency_mem.png -------------------------------------------------------------------------------- /assets/figures/mcunet_demo.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/mcunet/HEAD/assets/figures/mcunet_demo.gif -------------------------------------------------------------------------------- /assets/figures/memory_size.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/mcunet/HEAD/assets/figures/memory_size.png -------------------------------------------------------------------------------- /assets/figures/overview.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/mcunet/HEAD/assets/figures/overview.png -------------------------------------------------------------------------------- /assets/sample_images/person_det.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/mcunet/HEAD/assets/sample_images/person_det.jpg -------------------------------------------------------------------------------- /eval_det.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/mcunet/HEAD/eval_det.py -------------------------------------------------------------------------------- /eval_tflite.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/mcunet/HEAD/eval_tflite.py -------------------------------------------------------------------------------- /eval_torch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/mcunet/HEAD/eval_torch.py -------------------------------------------------------------------------------- /mcunet/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /mcunet/model_zoo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/mcunet/HEAD/mcunet/model_zoo.py -------------------------------------------------------------------------------- /mcunet/tinyengine/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/mcunet/HEAD/mcunet/tinyengine/README.md -------------------------------------------------------------------------------- /mcunet/tinynas/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /mcunet/tinynas/elastic_nn/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /mcunet/tinynas/elastic_nn/modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/mcunet/HEAD/mcunet/tinynas/elastic_nn/modules/__init__.py -------------------------------------------------------------------------------- /mcunet/tinynas/elastic_nn/modules/dynamic_layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/mcunet/HEAD/mcunet/tinynas/elastic_nn/modules/dynamic_layers.py -------------------------------------------------------------------------------- /mcunet/tinynas/elastic_nn/modules/dynamic_op.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/mcunet/HEAD/mcunet/tinynas/elastic_nn/modules/dynamic_op.py -------------------------------------------------------------------------------- /mcunet/tinynas/elastic_nn/networks/__init__.py: -------------------------------------------------------------------------------- 1 | from .ofa_proxyless import * 2 | -------------------------------------------------------------------------------- /mcunet/tinynas/elastic_nn/networks/ofa_proxyless.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/mcunet/HEAD/mcunet/tinynas/elastic_nn/networks/ofa_proxyless.py -------------------------------------------------------------------------------- /mcunet/tinynas/nn/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /mcunet/tinynas/nn/modules/__init__.py: -------------------------------------------------------------------------------- 1 | from .layers import * 2 | -------------------------------------------------------------------------------- /mcunet/tinynas/nn/modules/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/mcunet/HEAD/mcunet/tinynas/nn/modules/layers.py -------------------------------------------------------------------------------- /mcunet/tinynas/nn/networks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/mcunet/HEAD/mcunet/tinynas/nn/networks/__init__.py -------------------------------------------------------------------------------- /mcunet/tinynas/nn/networks/mobilenet_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/mcunet/HEAD/mcunet/tinynas/nn/networks/mobilenet_v2.py -------------------------------------------------------------------------------- /mcunet/tinynas/nn/networks/proxyless_nets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/mcunet/HEAD/mcunet/tinynas/nn/networks/proxyless_nets.py -------------------------------------------------------------------------------- /mcunet/tinynas/tf_codebase/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/mcunet/HEAD/mcunet/tinynas/tf_codebase/__init__.py -------------------------------------------------------------------------------- /mcunet/tinynas/tf_codebase/generate_tflite.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/mcunet/HEAD/mcunet/tinynas/tf_codebase/generate_tflite.py -------------------------------------------------------------------------------- /mcunet/tinynas/tf_codebase/tf_layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/mcunet/HEAD/mcunet/tinynas/tf_codebase/tf_layers.py -------------------------------------------------------------------------------- /mcunet/tinynas/tf_codebase/tf_model_zoo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/mcunet/HEAD/mcunet/tinynas/tf_codebase/tf_model_zoo.py -------------------------------------------------------------------------------- /mcunet/tinynas/tf_codebase/tf_modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/mcunet/HEAD/mcunet/tinynas/tf_codebase/tf_modules.py -------------------------------------------------------------------------------- /mcunet/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/mcunet/HEAD/mcunet/utils/__init__.py -------------------------------------------------------------------------------- /mcunet/utils/bn_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/mcunet/HEAD/mcunet/utils/bn_utils.py -------------------------------------------------------------------------------- /mcunet/utils/common_tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/mcunet/HEAD/mcunet/utils/common_tools.py -------------------------------------------------------------------------------- /mcunet/utils/det_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/mcunet/HEAD/mcunet/utils/det_helper.py -------------------------------------------------------------------------------- /mcunet/utils/my_modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/mcunet/HEAD/mcunet/utils/my_modules.py -------------------------------------------------------------------------------- /mcunet/utils/net_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/mcunet/HEAD/mcunet/utils/net_config.py -------------------------------------------------------------------------------- /mcunet/utils/pytorch_modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/mcunet/HEAD/mcunet/utils/pytorch_modules.py -------------------------------------------------------------------------------- /mcunet/utils/pytorch_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/mcunet/HEAD/mcunet/utils/pytorch_utils.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/mcunet/HEAD/requirements.txt -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mit-han-lab/mcunet/HEAD/setup.py --------------------------------------------------------------------------------