yolov5学习笔记:TypeError: list indices must be integers or slices, not list

wuguipatiechui

于 2024-08-12 15:16:33 发布

阅读量528

点赞数 5

文章标签： YOLO 学习笔记

本文链接：https://blog.csdn.net/wuguipatiechui/article/details/141132509

版权

我自己添加BiFPN时遇到报错，发现问题出在添加BiFPN_Add2与BiFPN_Add3中

from models.common import (
    C3,
    C3SPP,
    C3TR,
    SPP,
    SPPF,
    Bottleneck,
    BottleneckCSP,
    C3Ghost,
    C3x,
    Classify,
    Concat,
    Contract,
    Conv,
    CrossConv,
    DetectMultiBackend,
    DWConv,
    DWConvTranspose2d,
    Expand,
    Focus,
    GhostBottleneck,
    GhostConv,
    Proto，
    BiFPN_Add2,
    BiFPN_Add3
)这里添加以上 后续不需要添加

if m in {
    Conv,
    GhostConv,
    Bottleneck,
    GhostBottleneck,
    SPP,
    SPPF,
    DWConv,
    MixConv2d,
    Focus,
    CrossConv,
    BottleneckCSP,
    C3,
    C3TR,
    C3SPP,
    C3Ghost,
    nn.ConvTranspose2d,
    DWConvTranspose2d,
    C3x,
    CBAM,
}: #BiFPN_Add2,
    # BiFPN_Add3