-
Notifications
You must be signed in to change notification settings - Fork 25
无法处理1920*1080的图片 #10
Copy link
Copy link
Open
Description
如题,网络包含多次下采样操作,导致张量的尺寸变化情况如下:
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匹配上。
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
No labels