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

fix folder names, refactoring

Browse Source 
move to yolov8m
use checkpoint.txt instead of yaml
This commit is contained in:
Alex Bezdieniezhnykh
2024-06-29 22:18:34 +03:00
parent 3fd72202f2
commit bb1dbfe1e7
10 changed files with 116 additions and 164 deletions
Download Patch File Download Diff File Expand all files Collapse all files
constants.py
+18 -4
View File
@@ -1,9 +1,23 @@
import os
from dto.annotationClass import AnnotationClass
current_dataset_dir = os.path.join('datasets', 'zombobase-current')
current_images_dir = os.path.join(current_dataset_dir, 'images')
current_labels_dir = os.path.join(current_dataset_dir, 'labels')
prefix = 'azaion-'
images = 'images'
labels = 'labels'
data_dir = '/azaion/data/raw'
data_images_dir = os.path.join(data_dir, images)
data_labels_dir = os.path.join(data_dir, labels)
processed_dir = '/azaion/data/processed'
processed_images_dir = os.path.join(processed_dir, images)
processed_labels_dir = os.path.join(processed_dir, labels)
datasets_dir = '/azaion/datasets'
models_dir = '/azaion/models'
annotation_classes = AnnotationClass.read_json()
prefix = 'zombobase-'
date_format = '%Y-%m-%d'
checkpoint_file = 'checkpoint.txt'
checkpoint_date_format = '%Y-%m-%d %H:%M:%S'