├── .flake8 ├── .gitignore ├── LICENSE ├── README.md ├── dipoorlet ├── __init__.py ├── __main__.py ├── deploy │ ├── __init__.py │ ├── deploy_atlas.py │ ├── deploy_base.py │ ├── deploy_default.py │ ├── deploy_imx.py │ ├── deploy_magicmind.py │ ├── deploy_rv.py │ ├── deploy_snpe.py │ ├── deploy_stpu.py │ ├── deploy_ti.py │ └── deploy_trt.py ├── dist_helper.py ├── forward_net.py ├── platform_settings.py ├── profiling.py ├── quantize.py ├── tensor_cali │ ├── __init__.py │ ├── basic_algorithm.py │ └── tensor_cali_base.py ├── utils.py └── weight_transform │ ├── __init__.py │ ├── ada_quant_layer.py │ ├── adaround.py │ ├── bias_correction.py │ ├── brecq.py │ ├── sparse_quant.py │ ├── sparse_quant_layer.py │ ├── update_bn.py │ ├── utils.py │ ├── weight_equalization.py │ └── weight_trans_base.py ├── example ├── .gitkeep ├── magicmind.md ├── rv.md ├── snpe.md ├── tensorrt.md └── ti.md ├── requirements.txt └── setup.py /.flake8: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Dipoorlet/HEAD/.flake8 -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- 1 | *.pyc 2 | dipoorlet.egg-info/ 3 | -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Dipoorlet/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Dipoorlet/HEAD/README.md -------------------------------------------------------------------------------- /dipoorlet/__init__.py: -------------------------------------------------------------------------------- 1 | __version__ = '0.1.0' 2 | -------------------------------------------------------------------------------- /dipoorlet/__main__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Dipoorlet/HEAD/dipoorlet/__main__.py -------------------------------------------------------------------------------- /dipoorlet/deploy/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Dipoorlet/HEAD/dipoorlet/deploy/__init__.py -------------------------------------------------------------------------------- /dipoorlet/deploy/deploy_atlas.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Dipoorlet/HEAD/dipoorlet/deploy/deploy_atlas.py -------------------------------------------------------------------------------- /dipoorlet/deploy/deploy_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Dipoorlet/HEAD/dipoorlet/deploy/deploy_base.py -------------------------------------------------------------------------------- /dipoorlet/deploy/deploy_default.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Dipoorlet/HEAD/dipoorlet/deploy/deploy_default.py -------------------------------------------------------------------------------- /dipoorlet/deploy/deploy_imx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Dipoorlet/HEAD/dipoorlet/deploy/deploy_imx.py -------------------------------------------------------------------------------- /dipoorlet/deploy/deploy_magicmind.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Dipoorlet/HEAD/dipoorlet/deploy/deploy_magicmind.py -------------------------------------------------------------------------------- /dipoorlet/deploy/deploy_rv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Dipoorlet/HEAD/dipoorlet/deploy/deploy_rv.py -------------------------------------------------------------------------------- /dipoorlet/deploy/deploy_snpe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Dipoorlet/HEAD/dipoorlet/deploy/deploy_snpe.py -------------------------------------------------------------------------------- /dipoorlet/deploy/deploy_stpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Dipoorlet/HEAD/dipoorlet/deploy/deploy_stpu.py -------------------------------------------------------------------------------- /dipoorlet/deploy/deploy_ti.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Dipoorlet/HEAD/dipoorlet/deploy/deploy_ti.py -------------------------------------------------------------------------------- /dipoorlet/deploy/deploy_trt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Dipoorlet/HEAD/dipoorlet/deploy/deploy_trt.py -------------------------------------------------------------------------------- /dipoorlet/dist_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Dipoorlet/HEAD/dipoorlet/dist_helper.py -------------------------------------------------------------------------------- /dipoorlet/forward_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Dipoorlet/HEAD/dipoorlet/forward_net.py -------------------------------------------------------------------------------- /dipoorlet/platform_settings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Dipoorlet/HEAD/dipoorlet/platform_settings.py -------------------------------------------------------------------------------- /dipoorlet/profiling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Dipoorlet/HEAD/dipoorlet/profiling.py -------------------------------------------------------------------------------- /dipoorlet/quantize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Dipoorlet/HEAD/dipoorlet/quantize.py -------------------------------------------------------------------------------- /dipoorlet/tensor_cali/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Dipoorlet/HEAD/dipoorlet/tensor_cali/__init__.py -------------------------------------------------------------------------------- /dipoorlet/tensor_cali/basic_algorithm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Dipoorlet/HEAD/dipoorlet/tensor_cali/basic_algorithm.py -------------------------------------------------------------------------------- /dipoorlet/tensor_cali/tensor_cali_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Dipoorlet/HEAD/dipoorlet/tensor_cali/tensor_cali_base.py -------------------------------------------------------------------------------- /dipoorlet/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Dipoorlet/HEAD/dipoorlet/utils.py -------------------------------------------------------------------------------- /dipoorlet/weight_transform/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Dipoorlet/HEAD/dipoorlet/weight_transform/__init__.py -------------------------------------------------------------------------------- /dipoorlet/weight_transform/ada_quant_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Dipoorlet/HEAD/dipoorlet/weight_transform/ada_quant_layer.py -------------------------------------------------------------------------------- /dipoorlet/weight_transform/adaround.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Dipoorlet/HEAD/dipoorlet/weight_transform/adaround.py -------------------------------------------------------------------------------- /dipoorlet/weight_transform/bias_correction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Dipoorlet/HEAD/dipoorlet/weight_transform/bias_correction.py -------------------------------------------------------------------------------- /dipoorlet/weight_transform/brecq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Dipoorlet/HEAD/dipoorlet/weight_transform/brecq.py -------------------------------------------------------------------------------- /dipoorlet/weight_transform/sparse_quant.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Dipoorlet/HEAD/dipoorlet/weight_transform/sparse_quant.py -------------------------------------------------------------------------------- /dipoorlet/weight_transform/sparse_quant_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Dipoorlet/HEAD/dipoorlet/weight_transform/sparse_quant_layer.py -------------------------------------------------------------------------------- /dipoorlet/weight_transform/update_bn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Dipoorlet/HEAD/dipoorlet/weight_transform/update_bn.py -------------------------------------------------------------------------------- /dipoorlet/weight_transform/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Dipoorlet/HEAD/dipoorlet/weight_transform/utils.py -------------------------------------------------------------------------------- /dipoorlet/weight_transform/weight_equalization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Dipoorlet/HEAD/dipoorlet/weight_transform/weight_equalization.py -------------------------------------------------------------------------------- /dipoorlet/weight_transform/weight_trans_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Dipoorlet/HEAD/dipoorlet/weight_transform/weight_trans_base.py -------------------------------------------------------------------------------- /example/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /example/magicmind.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Dipoorlet/HEAD/example/magicmind.md -------------------------------------------------------------------------------- /example/rv.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Dipoorlet/HEAD/example/rv.md -------------------------------------------------------------------------------- /example/snpe.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Dipoorlet/HEAD/example/snpe.md -------------------------------------------------------------------------------- /example/tensorrt.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Dipoorlet/HEAD/example/tensorrt.md -------------------------------------------------------------------------------- /example/ti.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Dipoorlet/HEAD/example/ti.md -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Dipoorlet/HEAD/requirements.txt -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ModelTC/Dipoorlet/HEAD/setup.py --------------------------------------------------------------------------------