├── .gitignore ├── .gitmodules ├── README.md ├── checkpoints └── .gitignore ├── configs └── ssd300_coco.py ├── deploy_weight └── .gitignore ├── requirements.txt ├── src ├── autotvm_tuning.py ├── constants.py ├── convert_ssd_to_relay.py ├── custom_ops │ ├── .gitignore │ ├── custom_ops.cc │ └── load_custom_ops.py ├── deploy │ ├── .gitignore │ ├── CMakeLists.txt │ ├── README.md │ ├── main.cc │ ├── ssd_model.h │ └── tvm_runtime_pack.cc ├── modules │ └── traceable_ssd_module.py └── run_exported_ssd.py ├── test_images └── demo.jpg └── tuning_log └── .gitignore /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/randxie/mmdetection-tvm/HEAD/.gitignore -------------------------------------------------------------------------------- /.gitmodules: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/randxie/mmdetection-tvm/HEAD/.gitmodules -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/randxie/mmdetection-tvm/HEAD/README.md -------------------------------------------------------------------------------- /checkpoints/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/randxie/mmdetection-tvm/HEAD/checkpoints/.gitignore -------------------------------------------------------------------------------- /configs/ssd300_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/randxie/mmdetection-tvm/HEAD/configs/ssd300_coco.py -------------------------------------------------------------------------------- /deploy_weight/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/randxie/mmdetection-tvm/HEAD/deploy_weight/.gitignore -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | numpy 2 | onnx 3 | xgboost 4 | tornado -------------------------------------------------------------------------------- /src/autotvm_tuning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/randxie/mmdetection-tvm/HEAD/src/autotvm_tuning.py -------------------------------------------------------------------------------- /src/constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/randxie/mmdetection-tvm/HEAD/src/constants.py -------------------------------------------------------------------------------- /src/convert_ssd_to_relay.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/randxie/mmdetection-tvm/HEAD/src/convert_ssd_to_relay.py -------------------------------------------------------------------------------- /src/custom_ops/.gitignore: -------------------------------------------------------------------------------- 1 | build 2 | -------------------------------------------------------------------------------- /src/custom_ops/custom_ops.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/randxie/mmdetection-tvm/HEAD/src/custom_ops/custom_ops.cc -------------------------------------------------------------------------------- /src/custom_ops/load_custom_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/randxie/mmdetection-tvm/HEAD/src/custom_ops/load_custom_ops.py -------------------------------------------------------------------------------- /src/deploy/.gitignore: -------------------------------------------------------------------------------- 1 | build 2 | 3 | # ignore binaries 4 | run_ssd -------------------------------------------------------------------------------- /src/deploy/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/randxie/mmdetection-tvm/HEAD/src/deploy/CMakeLists.txt -------------------------------------------------------------------------------- /src/deploy/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/randxie/mmdetection-tvm/HEAD/src/deploy/README.md -------------------------------------------------------------------------------- /src/deploy/main.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/randxie/mmdetection-tvm/HEAD/src/deploy/main.cc -------------------------------------------------------------------------------- /src/deploy/ssd_model.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/randxie/mmdetection-tvm/HEAD/src/deploy/ssd_model.h -------------------------------------------------------------------------------- /src/deploy/tvm_runtime_pack.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/randxie/mmdetection-tvm/HEAD/src/deploy/tvm_runtime_pack.cc -------------------------------------------------------------------------------- /src/modules/traceable_ssd_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/randxie/mmdetection-tvm/HEAD/src/modules/traceable_ssd_module.py -------------------------------------------------------------------------------- /src/run_exported_ssd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/randxie/mmdetection-tvm/HEAD/src/run_exported_ssd.py -------------------------------------------------------------------------------- /test_images/demo.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/randxie/mmdetection-tvm/HEAD/test_images/demo.jpg -------------------------------------------------------------------------------- /tuning_log/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/randxie/mmdetection-tvm/HEAD/tuning_log/.gitignore --------------------------------------------------------------------------------