| 176 | return ori_img, img, word_vecs, mask, pad_masks, seg_id, sents |
| 177 | |
| 178 | def convert(self, img, mask, sent, inference=False): |
| 179 | img = Image.fromarray(np.uint8(img)) |
| 180 | mask = Image.fromarray(np.uint8(mask), mode="P") |
| 181 | img = F.resize(img, self.input_size) |
| 182 | if not inference: |
| 183 | mask = F.resize(mask, self.input_size, interpolation=Image.NEAREST) |
| 184 | img = F.to_tensor(img) |
| 185 | mask = torch.as_tensor(np.asarray(mask).copy(), dtype=torch.int64) |
| 186 | img = F.normalize(img, mean=self.mean, std=self.std) |
| 187 | return img, mask, sent |
| 188 | |
| 189 | def __repr__(self): |
| 190 | return self.__class__.__name__ + "(" + \ |