In order to realize the automatic optimization of hyperparameters of YOLO model, the hyperparameter optimization of you only look once (YOLO) model based on orthogonal optimization strategy (OOS) was proposed. Firstly, based on the principle of statistical orthogonal test, the orthogonal search method of population and the hyperparameter contribution analysis strategy were proposed to improve the optimization efficiency of the algorithm. Then, the uniform orthogonal search strategy and the neighborhood orthogonal search strategy were designed to alleviate the problem of the YOLO model falling into the local optimum and premature convergence. Finally, YOLOv5, YOLOv5s-Transformer and YOLOv7 were used as optimization objects to test on two target detection datasets, NWPU VHR-10 and Pascal VOC. Test results show that the recognition accuracy of the YOLO model is improved by the OOS hyperparameter optimization method in all cases. The average recognition accuracy mAP@0.5 on two datasets is improved to 93.94%, 93.18%, 93.45%, and 85.81%, 84.59%, 89.96%. The mAP@0.5-0.95 is improved to 60.00%, 60.08%, 56.98%,and 62.27%, 58.89%, 70.77%. It can provide a new intelligent method for hyperparameter optimization of object detection model.
| 科 Family | 属数 Number of genus | 种数 Number of species | 占总种数比例 Percentage of total species (%) | 属 Genus | 种数 Number of species | 占总种数比例 Percentage of total species (%) |
|---|---|---|---|---|---|---|
| 鹅膏菌科Amanitaceae | 2 | 11 | 5.26 | 鹅膏菌属 Amanita | 10 | 4.78 |
| 小菇科 Mycenaceae | 2 | 12 | 5.74 | 丝盖伞属 Inocybe | 5 | 2.39 |
| 多孔菌科 Polyporaceae | 8 | 14 | 6.70 | 蜡蘑属 Laccaria | 5 | 2.39 |
| 红菇科 Russulaceae | 3 | 23 | 11.00 | 小皮伞属 Marasmius | 6 | 2.87 |
| 小菇属 Mycena | 11 | 5.26 | ||||
| 光柄菇属 Pluteus | 5 | 2.39 | ||||
| 红菇属 Russula | 17 | 8.13 | ||||
| 栓菌属 Trametes | 5 | 2.39 |