142c6c4de8
Refactor constants management to use Pydantic BaseModel for configuration
...
- 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.
2026-03-27 18:18:30 +02:00
44c9e87bd4
remove unnecessary middle epochs
2025-05-31 19:02:57 +03:00
538dc8efa9
train is ready
...
manual_run reuses train's export
add current_model to constants
2025-05-31 18:41:10 +03:00
6e5153ffb7
prepare train.py to automated training
...
set cryptography lib to the certain version
add manual_run for manual operations, right now it is onnx conversion and upload
2025-05-31 14:50:58 +03:00
5b90320e27
train fixes merge
2025-05-22 17:02:24 +03:00
8fb24384db
add resume training possibility
2025-05-19 18:32:41 +03:00
96056f53ad
remove fog and shadow augmentations
...
add install script
2025-05-19 08:30:25 +03:00
292809593c
refactor augmentation to class, update classes.json, fix small bugs
2025-05-15 06:45:25 +03:00
26eb3b991d
add download big engine file to cdn manager
...
revise onnx export process
fixes
2025-04-24 16:23:39 +03:00
06a23525a6
fix tensor rt engine
2025-04-06 18:45:06 +03:00
5b89a21b36
add export to FP16
...
add inference with possibility to have different
2025-03-28 12:54:25 +02:00
eaef1a9b66
Merge remote-tracking branch 'origin/main'
...
# Conflicts:
# exports/export.py
2025-03-26 17:20:42 +02:00
3e02be7068
add export to tensorrt
2025-03-26 17:19:21 +02:00
7d99e377f1
add onnx inference
2025-03-24 21:48:22 +02:00
3113d59a3a
add weather modes
...
add exports
2025-03-13 16:39:57 +02:00
948d4f1425
simplify exports. part1
2025-03-13 16:10:23 +02:00
ee370fec05
add conversion to rockchip
2025-03-12 01:11:35 +02:00
458d59753a
GPU augmentation. try3. Failed.
2025-03-05 15:32:19 +02:00
b5e5f0b297
correct albumentation
...
try to make augmentation on GPU.
saved llm prompt
2025-03-05 10:45:41 +02:00
2fa864018f
upload model to cdn and api
...
switch to yolov11
2025-03-03 23:36:10 +02:00
ceb50bf48a
add rknn conversion - install and use scripts, auto convert to rknn after AI training is done and put pt and rknn models to /azaion/models directory
2025-02-21 14:26:54 +02:00
31c44943e8
add rknn conversion - install and use scripts, auto convert to rknn after AI training is done and put pt and rknn models to /azaion/models directory
2024-10-03 11:41:22 +03:00
c234e8b190
fix augmentation - correct bboxes to be within borders+margin
...
Add classes = 80 as default number in yaml file
2024-09-24 12:56:05 +03:00
543cead191
refactoring, fixes
2024-08-23 13:47:54 +03:00
6b0a0e678e
copy images and labels during forming dataset. add folder for corrupted labels, small refactor
2024-08-13 01:57:42 +03:00
bb1dbfe1e7
fix folder names, refactoring
...
move to yolov8m
use checkpoint.txt instead of yaml
2024-06-29 22:18:34 +03:00
3fd72202f2
add switch to save each epoch
2024-06-19 23:16:45 +03:00
eba8b62db8
train.py fix
2024-06-19 03:47:57 +03:00
6c50dd19b7
add dataset-visualiser.py
2024-06-19 02:28:50 +03:00
66987f4d95
add checkpoints and config system
...
convert from bbox oriented and pascal xml
fixes
2024-06-18 21:32:15 +03:00
2325fd0916
update gitignore
...
simplify paths
remove runs folder
2024-06-10 09:07:13 +03:00
70abaded05
small refactoring
...
simplify learning, improve readme
2024-06-09 18:09:26 +03:00
a9b777acc4
small refactoring
2024-06-08 19:34:17 +03:00
56017f2c23
add readme
...
update train.py with yolov10
fix generation of data.yaml
2024-06-08 19:32:04 +03:00
1f7b4a95d2
small fix
...
add code for revert dataset
add get_recent_model
2024-06-06 01:46:48 +03:00
07ea67746a
add train.py
...
form dataset for current date
add exception catching
2024-06-05 23:35:06 +03:00
3fd726f9c7
reorganizing files
...
add train
some catches
2024-05-29 23:24:58 +03:00
fb7ccdd2d8
Revert "Made some changes"
...
This reverts commit 8529bd0c6f
2024-05-26 23:10:28 +03:00
fdb784b392
Revert "Add check.py"
...
This reverts commit 6adc4c37e9
2024-05-26 23:10:25 +03:00
6adc4c37e9
Add check.py
2024-05-25 03:13:19 +03:00
8529bd0c6f
Made some changes
2024-05-23 17:27:36 +03:00
Oleksandr Bezdieniezhnykh