X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=blobdiff_plain;f=grids.py;h=247c146d16140ed24cc636d1baf74dc21aeefbeb;hb=c9c9df43a6b97e5b3e81c8cf05d2f1b3010dea05;hp=b40d5321aaf834adaba8c64c4c9427f5b0615651;hpb=47fcc6b14bf217e7b73e1b7f440fc6aaf862122b;p=culture.git diff --git a/grids.py b/grids.py index b40d532..247c146 100755 --- a/grids.py +++ b/grids.py @@ -65,19 +65,6 @@ class Grids(problem.Problem): return x - def frame2img_(self, x, scale=15): - x = x.reshape(x.size(0), self.height, -1) - x = self.colors[x].permute(0, 3, 1, 2) - s = x.shape - x = x[:, :, :, None, :, None].expand(-1, -1, -1, scale, -1, scale) - x = x.reshape(s[0], s[1], s[2] * scale, s[3] * scale) - - x[:, :, :, torch.arange(0, x.size(3), scale)] = 0 - x[:, :, torch.arange(0, x.size(2), scale), :] = 0 - x = x[:, :, 1:, 1:] - - return x - def save_image( self, result_dir,