azaion/ai-training
1
0
Fork 0
mirror of https://github.com/azaion/ai-training.git synced 2026-04-22 10:46:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

add rknn conversion - install and use scripts, auto convert to rknn after AI training is done and put pt and rknn models to /azaion/models directory

Browse Source
This commit is contained in:
zxsanny
2024-10-03 11:41:22 +03:00
parent c234e8b190
commit 31c44943e8
8 changed files with 122 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files
preprocess-train.py
+6
View File
@@ -0,0 +1,6 @@
from preprocessing import preprocess_annotations
from train import train_dataset, convert2rknn
preprocess_annotations()
train_dataset('2024-10-01')
convert2rknn()