Commit Graph

24 Commits

Author SHA1 Message Date
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