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

本文主要是介绍YOLOV5更换转置卷积，助力涨点！，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

由于转置卷积是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 root directory
if str(ROOT) not in sys.path:sys.path.append(str(ROOT))  # add ROOT to PATH
if platform.system() != "Windows":ROOT &#

这篇关于YOLOV5更换转置卷积，助力涨点！的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！

---