add conversion to rockchip

This commit is contained in:
Alex Bezdieniezhnykh
2025-03-12 01:11:35 +02:00
parent edf685fabc
commit ee370fec05
2 changed files with 9 additions and 0 deletions
+7
View File
@@ -0,0 +1,7 @@
from ultralytics import YOLO
def export_rknn(model_path):
model = YOLO(model_path)
model.export(format="rknn", name="rk3588")
pass
+2
View File
@@ -20,6 +20,7 @@ from constants import (processed_images_dir,
prefix, date_format,
datasets_dir, models_dir,
corrupted_images_dir, corrupted_labels_dir, sample_dir)
from exports.rockchip import export_rknn
from security import Security
from utils import Dotdict
@@ -207,6 +208,7 @@ def convert2rknn():
model.export(format="onnx")
pass
def form_data_sample(size=300):
images = []
with scandir(processed_images_dir) as imd: