Files
ai-training/constants.py
T
Oleksandr Bezdieniezhnykh 07ea67746a add train.py
form dataset for current date
add exception catching
2024-06-05 23:35:06 +03:00

13 lines
439 B
Python

import os
from datetime import datetime
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')
annotation_classes = AnnotationClass.read_json()
prefix = 'zombobase-'
today_dataset = os.path.join('datasets', f'{prefix}{datetime.now():%Y-%m-%d}')