(self, code, shape=None)
| 59 | return img2code(self.model, img) |
| 60 | |
| 61 | def DecodeIds(self, code, shape=None): |
| 62 | if shape is None: |
| 63 | if isinstance(code, list): |
| 64 | code = torch.tensor(code, device=self.device) |
| 65 | s = sqrt_int(len(code.view(-1))) |
| 66 | assert s * s == len(code.view(-1)) |
| 67 | shape = (1, s, s) |
| 68 | code = code.view(shape) |
| 69 | out = code2img(self.model, code) |
| 70 | return out |
| 71 | |
| 72 | def read_img(self, path, img_size=256): |
| 73 | tr = transforms.Compose([ |
no test coverage detected