Files
ai-training/constants.py
T
Alex Bezdieniezhnykh 66987f4d95 add checkpoints and config system
convert from bbox oriented and pascal xml
fixes
2024-06-18 21:32:15 +03:00

10 lines
353 B
Python

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')
annotation_classes = AnnotationClass.read_json()
prefix = 'zombobase-'
date_format = '%Y-%m-%d'