mirror of
https://github.com/azaion/ai-training.git
synced 2026-04-22 11:16:35 +00:00
simplify exports. part1
This commit is contained in:
@@ -1,7 +0,0 @@
|
|||||||
from ultralytics import YOLO
|
|
||||||
|
|
||||||
|
|
||||||
def export_rknn(model_path):
|
|
||||||
model = YOLO(model_path)
|
|
||||||
model.export(format="rknn", name="rk3588")
|
|
||||||
pass
|
|
||||||
@@ -15,3 +15,4 @@ pyyaml
|
|||||||
boto3
|
boto3
|
||||||
msgpack
|
msgpack
|
||||||
rstream
|
rstream
|
||||||
|
netron
|
||||||
@@ -20,7 +20,6 @@ from constants import (processed_images_dir,
|
|||||||
prefix, date_format,
|
prefix, date_format,
|
||||||
datasets_dir, models_dir,
|
datasets_dir, models_dir,
|
||||||
corrupted_images_dir, corrupted_labels_dir, sample_dir)
|
corrupted_images_dir, corrupted_labels_dir, sample_dir)
|
||||||
from exports.rockchip import export_rknn
|
|
||||||
from security import Security
|
from security import Security
|
||||||
from utils import Dotdict
|
from utils import Dotdict
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user