CornerNet的输入输出形状

标签：

xs:

torch.Size([10, 3, 255, 255])

ys:

torch.Size([10, 80, 64, 64])
torch.Size([10, 80, 64, 64])
torch.Size([10, 128])
torch.Size([10, 128, 2])
torch.Size([10, 128, 2])
torch.Size([10, 128])
torch.Size([10, 128])
torch.Size([10, 80, 64, 64])
torch.Size([10, 80, 64, 64])
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
torch.Size([10, 1, 16, 16])
torch.Size([10, 1, 32, 32])
torch.Size([10, 1, 64, 64])

输出：

torch.Size([2, 80, 64, 64])
torch.Size([2, 80, 64, 64])
torch.Size([2, 80, 64, 64])
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
torch.Size([2, 80, 64, 64])
torch.Size([2, 80, 64, 64])
torch.Size([2, 80, 64, 64])
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
torch.Size([2, 1, 64, 64])
torch.Size([2, 1, 64, 64])
torch.Size([2, 1, 64, 64])
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
torch.Size([2, 1, 64, 64])
torch.Size([2, 1, 64, 64])
torch.Size([2, 1, 64, 64])
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
torch.Size([2, 2, 64, 64])
torch.Size([2, 2, 64, 64])
torch.Size([2, 2, 64, 64])
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
torch.Size([2, 2, 64, 64])
torch.Size([2, 2, 64, 64])
torch.Size([2, 2, 64, 64])
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
torch.Size([2, 1, 16, 16])
torch.Size([2, 1, 32, 32])
torch.Size([2, 1, 64, 64])
torch.Size([2, 1, 16, 16])
torch.Size([2, 1, 32, 32])
torch.Size([2, 1, 64, 64])
torch.Size([2, 1, 16, 16])
torch.Size([2, 1, 32, 32])
torch.Size([2, 1, 64, 64])