mirror of
https://github.com/azaion/ai-training.git
synced 2026-04-22 11:06:35 +00:00
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.
11 lines
136 B
Plaintext
11 lines
136 B
Plaintext
pytest>=7.0
|
|
albumentations
|
|
opencv-python-headless
|
|
numpy==1.26.4
|
|
onnxruntime
|
|
cryptography==44.0.2
|
|
msgpack
|
|
PyYAML
|
|
ultralytics
|
|
coremltools
|