Skip to content

在使用[shortcut]时from参数为正数时的Bug反馈 #5

@ZaoYum-Liao

Description

@ZaoYum-Liao

您好,我在使用densenet201-bifpn.cfg进行训练的时候遇到了一个问题,如下图所示:
image
当from参数为正数时候,会出现通道数不匹配的问题,我在对代码进行查阅时候发现shortcut层转换代码如下所示:
image
当shortcut层选择204层和shortcut层的上一层进行相加时,其输出通道数目filters = output_filters[int(mdef['from'])]应该等于第204层的输出通道数目,其输出应该是一个等于512的通道数,其中204层的详细配置如下图所示:
image
然而由于output_filters在初始化的时候将图像的3通道作为列表的第一个值,就会出现层数和cfg文件不匹配的情况,比如在cfg文件的第y个shortcut层选择第0层进行相加,此时的第0层应该是cfg文件中的第0层的输出,在cfg/densenet201-bifpn.cfg中就是如下图所示:
image
然而此时我们的filters = output_filters[int(mdef['from'])]输出的确实3....
语言有点混乱,不知是否理解。我对代码进行更改,更改如下图所示:
image
亲测改完之后训练就没有问题了.....因为yolov3.cfg文件中shortcut层的from值都是负数,所以这个bug在测试from值全是负数的时候是没有问题的....

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions