From 948d4f142582bbf795d9f43151d63da766c2cf07 Mon Sep 17 00:00:00 2001 From: Alex Bezdieniezhnykh Date: Thu, 13 Mar 2025 16:10:23 +0200 Subject: [PATCH] simplify exports. part1 --- exports/rockchip.py | 7 ------- requirements.txt | 3 ++- train.py | 1 - 3 files changed, 2 insertions(+), 9 deletions(-) delete mode 100644 exports/rockchip.py diff --git a/exports/rockchip.py b/exports/rockchip.py deleted file mode 100644 index a030a11..0000000 --- a/exports/rockchip.py +++ /dev/null @@ -1,7 +0,0 @@ -from ultralytics import YOLO - - -def export_rknn(model_path): - model = YOLO(model_path) - model.export(format="rknn", name="rk3588") - pass \ No newline at end of file diff --git a/requirements.txt b/requirements.txt index 33a0976..ee36d10 100644 --- a/requirements.txt +++ b/requirements.txt @@ -14,4 +14,5 @@ requests pyyaml boto3 msgpack -rstream \ No newline at end of file +rstream +netron \ No newline at end of file diff --git a/train.py b/train.py index 3bd4673..02424dc 100644 --- a/train.py +++ b/train.py @@ -20,7 +20,6 @@ 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