mirror of
https://github.com/azaion/ai-training.git
synced 2026-04-22 11:06:35 +00:00
bb1dbfe1e7
move to yolov8m use checkpoint.txt instead of yaml
24 lines
624 B
Python
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'
|