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