Commit Graph

11 Commits (c8c3363926b0fb84ca623e93f1d58389528723f2)

Author SHA1 Message Date
hanoch c8c3363926 Fixed randomizer gets integers only! 2024-08-15 15:46:47 +03:00
hanoch c1a580e621 comment 2024-08-15 15:40:01 +03:00
hanoch b82fdc0574 Add support using the "parent" directory in data.yaml as parent for the images folder 2024-08-08 14:44:50 +03:00
hanoch 5bb345f112 fix annotation txtx file creation 2024-08-07 18:03:48 +03:00
Kıvanç Tezören 55b90e1119
Add option to use YOLOv5 AP metric (#775)
* Add YOLOv5 metric option

* Inform if using v5 metric
2022-09-16 21:14:01 +03:00
AlexeyAB84 36ce6b2087 Use compute_loss_ota() if there is not loss_ota param or loss_ota==1 2022-08-16 02:10:07 +03:00
AlexeyAB84 711a16ba57 Added param loss_ota for hyp.yaml, to disable OTA for faster training 2022-08-09 07:48:28 +03:00
Mohammad Khoshbin b8956dd5a5
fix training with frozen layers (#378) 2022-08-02 17:55:28 +03:00
Mohammad Khoshbin cc42a206ef
fix cpu-only training CUDA error (#369) 2022-07-30 18:43:18 +03:00
Dhia_Oussayed 0d882e553e
Fixed a bug for Hyperparameters Evolution. (#344)
* Update hyp.scratch.custom.yaml

anchors parameter missing, evolution won't start without.

* Update train.py

Updated the hyperparameter evolution metadata variable to match the hyp.yaml files for the evolution to run successfully.

* Update hyp.scratch.p5.yaml

Added the anchors parameter, evolution don't start without it.

* Update hyp.scratch.p6.yaml

added the anchors parameter for the hyperparameter evolution.

* Update hyp.scratch.tiny.yaml

added the anchors parameter for the hyperparameters evolution

* Update hyp.scratch.custom.yaml

* Update hyp.scratch.tiny.yaml

* Update hyp.scratch.p5.yaml

* Update hyp.scratch.p6.yaml

* Update train.py

* Update train_aux.py
2022-07-28 22:09:52 +03:00
Kin-Yiu, Wong a1c6e04c7c
main code
code for inference
2022-07-06 23:23:27 +08:00