Skip to content

模型保存时是按那个指标的? #11

@guohaochen

Description

@guohaochen
      模型保存时是按那个指标的?`fi = fitness(np.array(results).reshape(1, -1))  # fitness_i = weighted combination of [P, R, mAP, F1]`

您好,我看过你的论文,觉得你的工作非常好。其中我注意到,你的工作中只关注了F1 score这一个指标,恰好我的工作中也使用到这个指标,所以想问问你一个问题。 yolov5中保存模型会计算一个最优值,计算的是map0.5和map的加权和
fi = fitness(np.array(results).reshape(1, -1)
w = [0.0, 0.0, 0.1, 0.9] # weights for [P, R, mAP@0.5, mAP@0.5:0.95]
我想请问下你的工作保存模型是以那个指标来进行保存的,我看到你的注释中写道
# weighted combination of [P, R, mAP, F1]
请问是map和F1的加权和吗,权重比例是多少,是0.1 * map+0.9 * F1吗? 还是只取最大的F1 score?

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions