Commit Graph

7 Commits (55b90e111984dd85e7eed327e9ff271222aa8b82)

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