Commit Graph

127 Commits

Author SHA1 Message Date
zxsanny 292809593c refactor augmentation to class, update classes.json, fix small bugs 2025-05-15 06:45:25 +03:00
Alex Bezdieniezhnykh fd4cd75265 conversion to model with batch 4 is DONE 2025-04-24 16:24:15 +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
Alex Bezdieniezhnykh c4732cec8b add download big engine file to cdn manager
revise onnx export process
fixes
2025-04-24 16:23:34 +03:00
Alex Bezdieniezhnykh 9017fa7469 update cdn manager
download big engine file
2025-04-18 13:13:19 +03:00
Dennis Popov adcbdb4915 update cdn manager WIP 2025-04-18 10:39:05 +02:00
Alex Bezdieniezhnykh 3985d424c0 update requirements.txt 2025-04-06 18:48:24 +03:00
Alex Bezdieniezhnykh 22f7966ae9 Merge remote-tracking branch 'origin/main'
# Conflicts:
#	requirements.txt
2025-04-06 18:47:35 +03:00
zxsanny 06a23525a6 fix tensor rt engine 2025-04-06 18:45:06 +03:00
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