kneron-mmdetection is a platform built upon the well-known mmdetection for detection and instance segmentation. We encourage you to start with YOLOX: Step-By-Step to build basic knowledge of Kneron-Edition mmdetection, and read mmdetection docs for detailed mmdetection usage.
In this repository, we provide an end-to-end training/deployment flow to realize on Kneron's AI accelerators:
- Training/Evalulation:
- Modified model configuration file and verified for Kneron hardware platform
- Please see Overview of Benchmark and Model Zoo for Kneron-Verified model list
- Converting to ONNX:
- pytorch2onnx_kneron.py (beta)
- Export optimized and Kneron-toolchain supported onnx
- Automatically modify model for arbitrary data normalization preprocess
- Evaluation
- test_kneron.py (beta)
- Evaluate the model with pytorch checkpoint, onnx, and kneron-nef
- Testing
- inference_kn (beta)
- Verify the converted NEF model on Kneron USB accelerator with this API
- Converting Kneron-NEF: (toolchain feature)
- Convert the trained pytorch model to Kneron-NEF model, which could be used on Kneron hardware platform.
This project is released under the Apache 2.0 license.
N/A
| Backbone | size | Mem (GB) | box AP | Config | Download |
|---|---|---|---|---|---|
| YOLOX-s | 640 | 7.6 | 40.5 | config | model |
- Please refer to YOLOX: Step-By-Step, Step 0. Environment for installation.
- Please refer to Kneron PLUS - Python: Installation for the environment setup for Kneron USB accelerator.
- YOLOX: Step-By-Step: A tutorial for users to get started easily. To see detailed documents, please see below.
- [Kneron ONNX Export] (under development)
- [Kneron Inference] (under development)
- Kneron Toolchain Step-By-Step (YOLOv3)
- Kneron Toolchain Manual
- Original mmdetection getting started: It is recommended to read the original mmdetection getting started documents for other mmdetection operations.
- Original mmdetection readthedoc: Original mmdetection documents.
kneron-mmdetection a platform built upon OpenMMLab-mmdetection
-
For issues regarding to the original mmdetection: We appreciate all contributions to improve OpenMMLab-mmdetection. Ongoing projects can be found in out GitHub Projects. Welcome community users to participate in these projects. Please refer to CONTRIBUTING.md for the contributing guideline.
-
For issues regarding to this repository kneron-mmdetection: Welcome to leave the comment or submit pull requests here to improve kneron-mmdetection
- kneron-mmsegmentation: Kneron training/deployment platform on OpenMMLab - mmSegmentation semantic segmentation toolbox