mirror of
https://github.com/azaion/ai-training.git
synced 2026-04-22 09:06:35 +00:00
Oleksandr Bezdieniezhnykh
142c6c4de8
- Replaced module-level path variables in constants.py with a structured Pydantic Config class. - Updated all relevant modules (train.py, augmentation.py, exports.py, dataset-visualiser.py, manual_run.py) to access paths through the new config structure. - Fixed bugs related to image processing and model saving. - Enhanced test infrastructure to accommodate the new configuration approach. This refactor improves code maintainability and clarity by centralizing configuration management.
33 lines
565 B
YAML
33 lines
565 B
YAML
api:
|
|
url: 'https://api.azaion.com'
|
|
email: 'uploader@azaion.com'
|
|
password: 'Az@1on_10Upl0@der'
|
|
|
|
queue:
|
|
host: '188.245.120.247'
|
|
port: 5552
|
|
consumer_user: 'azaion_receiver'
|
|
consumer_pw: 'Az1onRecce777ve2r'
|
|
name: 'azaion-annotations'
|
|
|
|
dirs:
|
|
root: '/azaion'
|
|
data: 'data'
|
|
data_seed: 'data-seed'
|
|
data_processed: 'data-processed'
|
|
data_deleted: 'data_deleted'
|
|
images: 'images'
|
|
labels: 'labels'
|
|
|
|
training:
|
|
model: 'yolo11m.yaml'
|
|
epochs: 120
|
|
batch: 11
|
|
imgsz: 1280
|
|
save_period: 1
|
|
workers: 24
|
|
|
|
export:
|
|
onnx_imgsz: 1280
|
|
onnx_batch: 4
|