Skip to content

Wrong stride in yolo-s-mish model #11

@Darkhunter9

Description

@Darkhunter9

Hi:
It seems that the strids for yolo-s-mish model should be [16, 32, 64], instead of [8, 16, 32]. Referring to this:
https://github.com/LongxingTan/Yolov5/blob/88acfd988decc4cc78335cfb6eb50f1975294c1f/yolo/configs/yolo-s-mish.yaml#L15
Focus layer will leads to a decrease in w and h by 2, thus the output of this layer is of size bs * 208 * 208 * 32 instead of bs * 416 * 416 * 32.
Let me know if I'm wrong. Thanks!

Metadata

Metadata

Assignees

No one assigned

    Labels

    questionFurther information is requested

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions