mirror of
https://github.com/azaion/ai-training.git
synced 2026-04-22 08:56:35 +00:00
zxsanny
42 lines
1.2 KiB
Python
42 lines
1.2 KiB
Python
from os import path
|
|
|
|
azaion = '/azaion'
|
|
prefix = 'azaion-'
|
|
images = 'images'
|
|
labels = 'labels'
|
|
|
|
data_dir = path.join(azaion, 'data')
|
|
data_seeds_dir = path.join(azaion, 'data-seeds')
|
|
data_rabbit_dir = path.join(azaion, 'data-rabbit')
|
|
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'
|
|
ANNOTATIONS_QUEUE = 'azaion-annotations'
|
|
ANNOTATIONS_CONFIRMED_QUEUE = 'azaion-annotations-confirm'
|
|
OFFSET_FILE = 'offset.yaml'
|
|
|
|
AI_ONNX_MODEL_FILE_BIG = "azaion.onnx.big"
|
|
AI_ONNX_MODEL_FILE_SMALL = "azaion.onnx.small"
|
|
|
|
AI_TENSOR_MODEL_FILE_BIG = "azaion.engine.big"
|
|
AI_TENSOR_MODEL_FILE_SMALL = "azaion.engine.small"
|