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

copy images and labels during forming dataset. add folder for corrupted labels, small refactor

Browse Source
This commit is contained in:
zxsanny
2024-08-13 01:57:42 +03:00
parent bb1dbfe1e7
commit 6b0a0e678e
4 changed files with 120 additions and 80 deletions
Download Patch File Download Diff File Expand all files Collapse all files
constants.py
+9 -4
View File
@@ -1,21 +1,26 @@
import os
from dto.annotationClass import AnnotationClass
azaion = '/azaion'
prefix = 'azaion-'
images = 'images'
labels = 'labels'
data_dir = '/azaion/data/raw'
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)
processed_dir = '/azaion/data/processed'
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)
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)
datasets_dir = '/azaion/datasets'
models_dir = '/azaion/models'
datasets_dir = os.path.join(azaion, 'datasets')
models_dir = os.path.join(azaion, 'models')
annotation_classes = AnnotationClass.read_json()
date_format = '%Y-%m-%d'