Skip to content

无法处理1920*1080的图片 #10

@tzy2003

Description

@tzy2003

如题,网络包含多次下采样操作,导致张量的尺寸变化情况如下:
x shape: torch.Size([1, 3, 1920, 1080])
d1 shape: torch.Size([1, 64, 960, 540])
d2 shape: torch.Size([1, 128, 480, 270])
d3 shape: torch.Size([1, 256, 240, 135])
d4 shape: torch.Size([1, 512, 120, 67])
容易发现,由于1080无法被16整除,导致最后一次下采样会出现尺寸的四舍五入(实际上把0.5舍去了),进而导致后续上采样操作后的结果的最后一维是134,无法和d3的135匹配上。

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions