-
Notifications
You must be signed in to change notification settings - Fork 648
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
Add rv1126 yolov3 support to sdk #1280
Conversation
2442686
to
1407bd0
Compare
I have to do serval modifications when converting yolov3 torch model to rknn-int8 model.
I don't think users are willing to buy it. Let's try to eliminate the manual part.
|
Codes now automatically move the Normalization part for quantization to the rknn model. Tried to make codes automatically tackle the partition part but failed. The expression was not elegant and it would destroy the rule built before. |
How about deriving another task from |
Why? Actually, I also used other backends with partition configs, which is convenient for debugging. These codes are more related to partition settings than just the RKNN backend. |
I found the API of
Make |
@grimoire Now, the process is moved to the base task. New arguments for |
* add yolov3 head to SDK * add yolov5 head to SDK * fix export-info and lint, add reverse check * fix lint * fix export info for yolo heads * add output_names to partition_config * fix typo * config * normalize config * fix * refactor config * fix lint and doc * c++ form * resolve comments * fix CI * fix CI * fix CI * float strides anchors * refine pipeline of rknn-int8 * config * rename func * refactor * rknn wrapper dict and fix typo * rknn wrapper output update, mmcls use end2end type * fix typo
After #1203 and #1238