(x)
| 45 | |
| 46 | |
| 47 | def array2grid(x): |
| 48 | nrow = round(math.sqrt(x.size(0))) |
| 49 | x = make_grid(x.clamp(0, 1), nrow=nrow, value_range=(0, 1)) |
| 50 | x = x.mul(255).add_(0.5).clamp_(0, 255).permute(1, 2, 0).to('cpu', torch.uint8).numpy() |
| 51 | return x |
| 52 | |
| 53 | |
| 54 | @torch.no_grad() |