YOLOV5更换转置卷积，助力涨点！

Dandelion701

于 2024-05-09 17:20:21 发布

阅读量252

点赞数 1

CC 4.0 BY-SA版权

分类专栏： yoloV5改进，涨点文章标签： YOLO 深度学习人工智能

本文链接：https://blog.csdn.net/weixin_57058018/article/details/138622250

yoloV5改进，涨点专栏收录该内容

15 篇文章 ¥15.90 ¥99.00

订阅专栏

超级会员免费看

本文介绍了如何在YOLOV5中替换转置卷积操作，通过修改yolov5n.yaml配置文件并更新yolo.py，实现模型改进。完成替换后，模型得以更新，遇到问题可在评论区讨论。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

由于转置卷积是nn库自带的，所以我们直接找到models文件夹中的yolo.py文件中的
parse_model函数，再在如下图的地方添加转置卷积模块

# YOLOv5 🚀 by Ultralytics, AGPL-3.0 license
"""
YOLO-specific modules.

Usage:
    $ python models/yolo.py --cfg yolov5s.yaml
"""

import argparse
import contextlib
import math
import os
import platform
import sys
from copy import deepcopy
from pathlib import Path
import torch
import torch.nn as nn
from models.FLAttention import *
FILE = Path(__file__).resolve()
ROOT = FILE.parents[1]  # YOLOv5 roo