Как извлечь конкретные показатели (mAP) из функции поезда YOLOv7? ⇐ Python

[Anonymous]

Я использую функцию поезда из файла: https://github.com/WongKinYiu/yolov7/bl ... ain_aux.py для обучения модели YOLOv7 на пользовательском наборе данных. Я хотел бы оптимизировать гиперпараметры (с помощью Optuna), чтобы добиться оптимальной производительности модели.
Я могу генерировать различные значения гиперпараметров с помощью optuna.trial.Trial импортируйте правильную функцию поезда и вызовите ее с этими гиперпараметрами. Я застрял в извлечении mAP@@0.5:0.95 из возвращаемого значения функции поезда, чтобы я мог вернуть его из целевой функции.
Я попробовал используя подпроцесс для вызова файла train_aux.py с разными гиперпараметрами и регулярным выражением для извлечения метрики mAP@@0.5:0.95 для оптимизации, но это был неправильный способ сделать это.

Подробнее здесь: https://stackoverflow.com/questions/792 ... n-function