NameError: name ‘DWConvTranspose2d‘ is not defined【yolov5自训练报错日志及解决】

wenzi404

于 2023-07-17 23:27:07 发布

阅读量844

点赞数

分类专栏： yolov5自训练文章标签： YOLO 深度学习人工智能

本文链接：https://blog.csdn.net/owenzi0/article/details/131777085

版权

yolov5自训练专栏收录该内容

10 篇文章

订阅专栏

在EfficientVIT改进版的YOLOv5训练过程中遇到NameError，提到DWConvTranspose2d未定义。为解决此问题，文章建议在common.py文件中创建DWConvTranspose2d类，实现深度卷积转置层，初始化方法中设置groups参数为输入和输出通道的最大公约数。

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

模型前提：采用Efficient VIT改进yolov5

Q：运行训练时报错：

NameError: name 'DWConvTranspose2d' is not defined

S：在common.py下创建该类：

class DWConvTranspose2d(nn.ConvTranspose2d):
    # Depth-wise transpose convolution
    def __init__(self, c1, c2, k=1, s=1, p1=0, p2=0):  # ch_in, ch_out, kernel, stride, padding, padding_out
        super().__init__(c1, c2, k, s, p1, p2, groups=math.gcd(c1, c2))