U版YOLOv5学习笔记之训练参数

最新推荐文章于 2024-07-25 16:59:12 发布

Jekk_cheng

最新推荐文章于 2024-07-25 16:59:12 发布

阅读量8k

点赞数 8

分类专栏： YOLO学习笔记文章标签：深度学习计算机视觉 pytorch 神经网络

本文链接：https://blog.csdn.net/qq_34437210/article/details/106821490

版权

源码链接：https://github.com/ultralytics/yolov5/blob/master/train.py

使用字典存储模型的超参数

# Hyperparameters
hyp = {
   'lr0': 0.01,  # initial learning rate (SGD=1E-2, Adam=1E-3)
       'momentum': 0.937,  # SGD momentum
       'weight_decay': 5e-4,  # optimizer weight decay
       'giou': 0.05,  # giou loss gain
       'cls': 0.58,  # cls loss gain
       'cls_pw': 1.0,  # cls BCELoss positive_weight
       'obj': 1.0,  # obj loss gain (*=img_size/320 if img_size != 320)
       'obj_pw': 1.0,  # obj BCELoss positive_weight
       'iou_t': 0.20,  # iou training threshold
       'anchor_t': 4.0,  # anchor-multiple threshold
       'fl_gamma': 0.0,  # focal loss gamma (efficientDet default is gamma=1.5)
       'hsv_h': 0.014,  # image HSV-Hue augmentation (fraction)
       'hsv_s': 0.68,  # image HSV