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