Convert a batch of code to imgs Args: model: ... code: [b, h, w] or [b, h*w] LongTensor
(model, code)
| 223 | return id_t1.view(img.shape[0], -1) |
| 224 | |
| 225 | def code2img(model, code): |
| 226 | '''Convert a batch of code to imgs |
| 227 | Args: |
| 228 | model: ... |
| 229 | code: [b, h, w] or [b, h*w] LongTensor |
| 230 | ''' |
| 231 | if len(code.shape) == 2: |
| 232 | s = int(math.sqrt(len(code.view(-1))) + 1e-5) |
| 233 | code = code.view(code.shape[0], s, s) |
| 234 | with torch.no_grad(): |
| 235 | out = model.decode_code(code) |
| 236 | out = out * torch.tensor([0.30379, 0.32279, 0.32800], device=out.device).view(1, -1, 1, 1) + torch.tensor([0.79093, 0.76271, 0.75340], device=out.device).view(1, -1, 1, 1) |
| 237 | return out |
| 238 |