azaion/ai-training
1
0
Fork 0
mirror of https://github.com/azaion/ai-training.git synced 2026-04-22 11:16:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

add checkpoints and config system

Browse Source 
convert from bbox oriented and pascal xml
fixes
This commit is contained in:
Alex Bezdieniezhnykh
2024-06-18 21:32:15 +03:00
parent b7b8b8fd27
commit 66987f4d95
6 changed files with 182 additions and 59 deletions
Download Patch File Download Diff File Expand all files Collapse all files
constants.py
+2 -2
View File
@@ -1,9 +1,9 @@
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-'
date_format = '%Y-%m-%d'