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

add onnx inference

Browse Source
This commit is contained in:
Alex Bezdieniezhnykh
2025-03-24 21:48:22 +02:00
parent 3113d59a3a
commit 7d99e377f1
7 changed files with 242 additions and 89 deletions
Download Patch File Download Diff File Expand all files Collapse all files
exports/export.py
+3 -2
View File
@@ -6,7 +6,6 @@ from ultralytics import YOLO
def export_rknn(model_path):
# model_onnx = export_onnx(model_path)
model = YOLO(model_path)
model.export(format="rknn", name="rk3588", simplify=True)
model_stem = Path(model_path).stem
@@ -25,10 +24,12 @@ def export_onnx(model_path):
nms=True)
return Path(model_path).stem + '.onnx'
def show_model(model: str = None):
netron.start(model)
if __name__ == '__main__':
show_model('azaion_2025-03-10.rknn')
export_onnx('azaion-2024-10-26.pt')
show_model('azaion-2024-10-26.onnx')
# export_rknn('azaion_2025-03-10.pt')