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

add rknn conversion - install and use scripts, auto convert to rknn after AI training is done and put pt and rknn models to /azaion/models directory

Browse Source
This commit is contained in:
zxsanny
2025-02-21 14:26:54 +02:00
parent 6e4b0f40ef
commit ceb50bf48a
11 changed files with 205 additions and 46 deletions
Download Patch File Download Diff File Expand all files Collapse all files
constants.py
+13 -12
View File
@@ -1,4 +1,4 @@
import os
from os import path
from dto.annotationClass import AnnotationClass
azaion = '/azaion'
@@ -6,21 +6,22 @@ prefix = 'azaion-'
images = 'images'
labels = 'labels'
data_dir = os.path.join(azaion, 'data')
data_images_dir = os.path.join(data_dir, images)
data_labels_dir = os.path.join(data_dir, labels)
data_dir = path.join(azaion, 'data')
data_images_dir = path.join(data_dir, images)
data_labels_dir = path.join(data_dir, labels)
processed_dir = os.path.join(azaion, 'data-processed')
processed_images_dir = os.path.join(processed_dir, images)
processed_labels_dir = os.path.join(processed_dir, labels)
processed_dir = path.join(azaion, 'data-processed')
processed_images_dir = path.join(processed_dir, images)
processed_labels_dir = path.join(processed_dir, labels)
corrupted_dir = os.path.join(azaion, 'data-corrupted')
corrupted_images_dir = os.path.join(corrupted_dir, images)
corrupted_labels_dir = os.path.join(corrupted_dir, labels)
corrupted_dir = path.join(azaion, 'data-corrupted')
corrupted_images_dir = path.join(corrupted_dir, images)
corrupted_labels_dir = path.join(corrupted_dir, labels)
sample_dir = path.join(azaion, 'data-sample')
datasets_dir = os.path.join(azaion, 'datasets')
models_dir = os.path.join(azaion, 'models')
datasets_dir = path.join(azaion, 'datasets')
models_dir = path.join(azaion, 'models')
annotation_classes = AnnotationClass.read_json()
date_format = '%Y-%m-%d'