From c9c9df43a6b97e5b3e81c8cf05d2f1b3010dea05 Mon Sep 17 00:00:00 2001 From: =?utf8?q?Fran=C3=A7ois=20Fleuret?= Date: Sat, 6 Jul 2024 10:05:00 +0300 Subject: [PATCH] Update. --- grids.py | 13 ------------- 1 file changed, 13 deletions(-) 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, -- 2.39.5