mirror of
https://github.com/azaion/ai-training.git
synced 2026-04-22 22:26:36 +00:00
Alex Bezdieniezhnykh
43 lines
1.1 KiB
Python
43 lines
1.1 KiB
Python
from os import path
|
|
|
|
azaion = '/azaion'
|
|
prefix = 'azaion-'
|
|
images = 'images'
|
|
labels = 'labels'
|
|
|
|
data_dir = path.join(azaion, 'data')
|
|
data_images_dir = path.join(data_dir, images)
|
|
data_labels_dir = path.join(data_dir, labels)
|
|
|
|
processed_dir = path.join(azaion, 'data-processed')
|
|
processed_images_dir = path.join(processed_dir, images)
|
|
processed_labels_dir = path.join(processed_dir, labels)
|
|
|
|
corrupted_dir = path.join(azaion, 'data-corrupted')
|
|
corrupted_images_dir = path.join(corrupted_dir, images)
|
|
corrupted_labels_dir = path.join(corrupted_dir, labels)
|
|
|
|
sample_dir = path.join(azaion, 'data-sample')
|
|
|
|
datasets_dir = path.join(azaion, 'datasets')
|
|
models_dir = path.join(azaion, 'models')
|
|
|
|
|
|
date_format = '%Y-%m-%d'
|
|
checkpoint_file = 'checkpoint.txt'
|
|
checkpoint_date_format = '%Y-%m-%d %H:%M:%S'
|
|
|
|
CONFIG_FILE = 'config.yaml'
|
|
|
|
JPG_EXT = '.jpg'
|
|
TXT_EXT = '.txt'
|
|
|
|
OFFSET_FILE = 'offset.yaml'
|
|
|
|
SMALL_SIZE_KB = 3
|
|
CDN_CONFIG = 'cdn.yaml'
|
|
MODELS_FOLDER = 'models'
|
|
|
|
CURRENT_PT_MODEL = path.join(models_dir, f'{prefix[:-1]}.pt')
|
|
CURRENT_ONNX_MODEL = path.join(models_dir, f'{prefix[:-1]}.onnx')
|