Commit 7a54c6be authored by Alykhan Tejani's avatar Alykhan Tejani Committed by Francisco Massa
Browse files

Fix uneven padding in make grid (#248)

* load densenet tensors onto cpu

* fix uneven padding in make_grid

* revert incorrectly added commit
parent 36849429
...@@ -70,14 +70,14 @@ def make_grid(tensor, nrow=8, padding=2, ...@@ -70,14 +70,14 @@ def make_grid(tensor, nrow=8, padding=2,
xmaps = min(nrow, nmaps) xmaps = min(nrow, nmaps)
ymaps = int(math.ceil(float(nmaps) / xmaps)) ymaps = int(math.ceil(float(nmaps) / xmaps))
height, width = int(tensor.size(2) + padding), int(tensor.size(3) + padding) height, width = int(tensor.size(2) + padding), int(tensor.size(3) + padding)
grid = tensor.new(3, height * ymaps + 1 + padding // 2, width * xmaps + 1 + padding // 2).fill_(pad_value) grid = tensor.new(3, height * ymaps + padding, width * xmaps + padding).fill_(pad_value)
k = 0 k = 0
for y in irange(ymaps): for y in irange(ymaps):
for x in irange(xmaps): for x in irange(xmaps):
if k >= nmaps: if k >= nmaps:
break break
grid.narrow(1, y * height + 1 + padding // 2, height - padding)\ grid.narrow(1, y * height + padding, height - padding)\
.narrow(2, x * width + 1 + padding // 2, width - padding)\ .narrow(2, x * width + padding, width - padding)\
.copy_(tensor[k]) .copy_(tensor[k])
k = k + 1 k = k + 1
return grid return grid
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment