- 
          
 - 
                Notifications
    
You must be signed in to change notification settings  - Fork 324
 
Support for Rockchip NPUs #1081
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
base: master
Are you sure you want to change the base?
Conversation
8e5c762    to
    31e19e4      
    Compare
  
    | 
           Vermin indicates that   | 
    
| 
           Things to discuss: 
  | 
    
| 
           Of note: this may also address #842, since the RK3588 has the same NPU series. I don't have any boards with that part, I can't confirm.  | 
    
| 
           Oh one thing I should have noticed sooner - the PR needs to use "master" branch as the base. We don't directly merge to the "release" branch. Can you change the base of this PR to "master"? Thank you!  | 
    
          
 Rebased to master, I missed it too.  | 
    
fb82959    to
    fa0d009      
    Compare
  
    | 
           TODO: 
  | 
    
| 
           TODO: 
  | 
    
This adds support for RKNN-based inference of the YOLO2-based model.
RKNN uses its own model format, which (in this case) is made by converting a preexisting ONNX one.
In addition to a new RKNNNet in ml_api, this also includes.
2 scripts for creating RKNN models:
convert_to_rknnconverts a single model from ONNX to RKNNmake_rknn_imagesconverts an ONNX model into a series of RKNN models, targeting different parts.Two new containers:
rknn_toolkit: Debian-based with the full rknn toolkit and a copy ofmake_rknn_images, meant for batch-converting easily.base_arm64_rknn: An arm64/aarch64 base with the RKNNLite runtime, and rknpu2 library preinstalled.TODOs: