Files
ai-training/constants.py
T
Alex Bezdieniezhnykh bb1dbfe1e7 fix folder names, refactoring
move to yolov8m
use checkpoint.txt instead of yaml
2024-06-29 22:18:34 +03:00

24 lines
624 B
Python

import os
from dto.annotationClass import AnnotationClass
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()
date_format = '%Y-%m-%d'
checkpoint_file = 'checkpoint.txt'
checkpoint_date_format = '%Y-%m-%d %H:%M:%S'