kneron-mmpose is a platform built upon the well-known mmpose for pose estimation . We encourage that you may start with the rsn18_hand_pose_step_by_step.md to build basic knowledge of Kneron-Edition MMPose, and read mmpose docs for detailed mmpose 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 and verified for Kneron hardware platform
- Please see Overview of Benchmark and Model Zoo for the model list
- Converting to onnx:
- pytorch2onnx_kneron.py (beta)
- Export optimized and Kneron-toolchain supported onnx
- Automatically modify model for arbitrary data normalization preprocess
- General Evaluation
- tools/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) | AUC | PCK | EPE | Config | Download |
|---|---|---|---|---|---|---|---|
| rsn18 | 224 | 0.014 | 0.855 | 0.987 | 3.687 | config | model |
- Please refer to get_started.md for installation.
- Please refer to Kneron PLUS - Python: Installation for the environment setup for Kneron USB accelerator.
- rsn18_hand_pose_step_by_step.md: 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 mmpose getting started: It is recommended to read original MMXXXX getting started documents for other mmpose operations.
- original mmpose readthedoc: Original mmpose documents.
kneron-mmpose a platform built upon OpenMMLab-mmpose
-
For issues regarding to the original mmpose: We appreciate all contributions to improve OpenMMLab-mmpose. 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-mmpose: Welcome to leave the comment or submit the pull request here to improve kneron-mmpose
- kneron-mmdetection: Kneron training/deployment platform on OpenMMLab - mmdetection object detection toolbox