azaion/ai-training
1
0
Fork 0
mirror of https://github.com/azaion/ai-training.git synced 2026-04-22 21:46:35 +00:00
Code Issues Packages Projects Releases Wiki Activity
68 Commits 3 Branches 0 Tags
aaa4e2c2d359468e4156cda3757f641121f9131c
Commit Graph

68 Commits

This Branch
This Branch
All Branches
Author SHA1 Message Date
zxsanny aaa4e2c2d3 update requirements 2025-03-28 14:50:43 +02: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
Alex Bezdieniezhnykh edf685fabc fix rabbit consumer 2025-03-07 14:33:02 +02:00
zxsanny 4e8c67995a read from queue 2025-03-06 18:34:05 +02:00
Alex Bezdieniezhnykh eb94311294 Merge remote-tracking branch 'origin/main' 
# Conflicts:
#	requirements.txt
 2025-03-05 21:06:41 +02:00
Alex Bezdieniezhnykh b4a4826b19 add rabbit consumer 2025-03-05 20:58:33 +02:00
zxsanny 458d59753a GPU augmentation. try3. Failed. 2025-03-05 15:32:19 +02:00
zxsanny 2611a22264 GPU augmentation. try2 2025-03-05 10:48:45 +02:00
zxsanny 5bd519b968 GPU augmentation. try1 2025-03-05 10:47:13 +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
Alex Bezdieniezhnykh 6e4b0f40ef update install.sh script 2024-10-16 17:10:33 +03:00
Alex Bezdieniezhnykh b23d1639ae add scripts for orange pi install all libs and run inference using rknn model 2024-10-04 00:14:46 +03:00
zxsanny d09e16b5a7 remove old script 2024-10-03 11:47:54 +03: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 42ea4ebc2f Merge remote-tracking branch 'origin/main' 2024-08-23 13:48:03 +03:00
zxsanny 543cead191 refactoring, fixes 2024-08-23 13:47:54 +03:00
Alex Bezdieniezhnykh 330d4b914a fix readme 
add export py
add dll
 2024-08-23 13:40:08 +03:00
Oleksandr Bezdieniezhnykh d2dae732b9 Merge remote-tracking branch 'origin/main' 2024-08-15 23:38:24 +03:00
Oleksandr Bezdieniezhnykh 72c86443f9 add test-prediction.py 2024-08-15 23:38:08 +03:00
zxsanny 9c9e3d8d44 Merge remote-tracking branch 'origin/main' 2024-08-13 01:58:01 +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
Oleksandr Bezdieniezhnykh 7cf85f66f1 Merge remote-tracking branch 'origin/main' 2024-08-12 17:54:18 +03:00
Oleksandr Bezdieniezhnykh ffda0fa7ef add init sftp script 2024-08-12 17:54:04 +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 165d609d01 add fixes for classes 2024-06-18 21:52:22 +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
Oleksandr Bezdieniezhnykh b7b8b8fd27 small refactoring 2024-06-16 12:21:38 +03:00
Oleksandr Bezdieniezhnykh 4161078f40 add convert-annotations.py for converting other dataset formats 2024-06-15 17:52:56 +03:00
Alex Bezdieniezhnykh 2325fd0916 update gitignore 
simplify paths
remove runs folder
 2024-06-10 09:07:13 +03:00
Alex Bezdieniezhnykh 3bd34134d2 update gitignore 2024-06-09 18:11: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 34d4fd9623 proper augmentations. Add 7 more images per 1 image 
make dirs if not exists
add visualize
 2024-05-28 16:39:38 +03:00
Oleksandr Bezdieniezhnykh 02b45f83d1 add more composition stepss 2024-05-27 03:16:46 +03:00