Yolov5修改损失函数

weixin_44119674

已于 2023-11-07 09:43:28 修改

阅读量993

点赞数 5

分类专栏：算法实战 deeep learning 文章标签： YOLO 深度学习 python

于 2023-11-07 09:39:25 首次发布

本文链接：https://blog.csdn.net/weixin_44119674/article/details/134260126

版权

deeep learning 同时被 2 个专栏收录

108 篇文章 7 订阅 ¥19.90 ¥99.00

订阅专栏

超级会员免费看

算法实战

44 篇文章 0 订阅 ¥19.90 ¥99.00

订阅专栏

超级会员免费看

本文介绍了如何在YOLOv5中修改损失函数，主要涉及train.py、val.py和loss.py三个文件的改动，详细阐述了每个文件中的关键调整步骤。

摘要由CSDN通过智能技术生成

Yolov5修改损失函数，主要是修改如下三个py文件：train.py、val.py、loss.py

一、train.py


# YOLOv5 🚀 by Ultralytics, GPL-3.0 license
"""
Train a YOLOv5 model on a custom dataset.
Models and datasets download automatically from the latest YOLOv5 release.

Usage - Single-GPU training:
    $ python train.py --data coco128.yaml --weights yolov5s.pt --img 640  # from pretrained (recommended)
    $ python train.py --data coco128.yaml --weights '' --cfg yolov5s.yaml --img 640  # from scratch

Usage - Multi-GPU DDP training:
    $ python -m torch.distributed.run --nproc_per_node 4 --master_port 1 train.py --data coco128.yaml --weights yolov5s.pt

了解本专栏