mirror of
https://github.com/azaion/ai-training.git
synced 2026-04-22 13:06:36 +00:00
add train.py
form dataset for current date add exception catching
This commit is contained in:
@@ -0,0 +1,13 @@
|
||||
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}')
|
||||
Reference in New Issue
Block a user