Commit Graph

97 Commits

Author SHA1 Message Date
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
Oleksandr Bezdieniezhnykh 4766269d3a delete breeding
add proper preprocessing. not so tested yet
2024-05-27 03:13:58 +03:00
Oleksandr Bezdieniezhnykh 831de7fea3 Revert "Made some changes"
This reverts commit ed1af3de97.
2024-05-26 23:10:37 +03:00
Oleksandr Bezdieniezhnykh 316e996d0d Revert "Made some changes"
This reverts commit c1d30d024d.
2024-05-26 23:10:31 +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
Oleksandr Bezdieniezhnykh f463a6fcad Revert "Add check.py"
This reverts commit f8b8747f5e.
2024-05-26 23:10:21 +03:00
Oleksandr Bezdieniezhnykh a1e08e640f Revert "Made some changes"
This reverts commit 98c604ff93.
2024-05-26 23:10:18 +03:00
Nazar Sturanec 98c604ff93 Made some changes 2024-05-25 03:26:39 +03:00
Nazar Sturanec f8b8747f5e Add check.py 2024-05-25 03:26:04 +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
Nazar Sturanec c1d30d024d Made some changes 2024-05-22 15:56:46 +03:00
Nazar Sturanec ed1af3de97 Made some changes 2024-05-22 15:55:31 +03:00
Oleksandr Bezdieniezhnykh e94f3e98c9 add structure how it should work
in preprocessing.py
2024-05-21 15:25:54 +03:00
Nazar Sturanec d5461d47e2 Made some changes 2024-05-21 13:53:17 +03:00
Nazar Sturanec a562f51f66 Add breeding_date_set.py 2024-05-20 16:27:38 +03:00
Oleksandr Bezdieniezhnykh 285c9e4266 add classes.json 2024-05-18 19:42:26 +03:00
Oleksandr Bezdieniezhnykh 5dbbd17ff1 initial commit
define folder structure, add example files
add gitignore
2024-05-18 19:40:02 +03:00