yolo5目标检测网络学习

yolo5的训练trick

https://zhuanlan.zhihu.com/p/197776260

ema

权重系数随着迭代次数的变化趋势如下,当迭代次数增加时,权重系数越来越大,则后续模型的参数对整体ema模型的参数影响越来越小

import numpy as np
    import matplotlib.pyplot as plt
    x = []
    y = []
    for i in range(1, 10000, 10):
        x.append([i])
        y.append(0.9999*(1-np.exp(i*-1/2000)))

    data_x = np.array(x)
    data_y = np.array(y)
    plt.ylabel("decay")
    plt.xlabel("batch")
    plt.plot(x, y)
    plt.show()
权重系数变化趋势

Random HSV

augment_hsv(img, hgain=hyp['hsv_h'], sgain=hyp['hsv_s'], vgain=hyp['hsv_v'])

 

posted @ 2024-09-21 13:08  silence_cho  阅读(24)  评论(0)    收藏  举报