Test with augmentations. Only rescale=True is supported.
(self, imgs, img_metas, rescale=True)
| 274 | return seg_pred |
| 275 | |
| 276 | def aug_test(self, imgs, img_metas, rescale=True): |
| 277 | """Test with augmentations. |
| 278 | |
| 279 | Only rescale=True is supported. |
| 280 | """ |
| 281 | # aug_test rescale all imgs back to ori_shape for now |
| 282 | assert rescale |
| 283 | # to save memory, we get augmented seg logit inplace |
| 284 | seg_logit = self.inference(imgs[0], img_metas[0], rescale) |
| 285 | for i in range(1, len(imgs)): |
| 286 | cur_seg_logit = self.inference(imgs[i], img_metas[i], rescale) |
| 287 | seg_logit += cur_seg_logit |
| 288 | seg_logit /= len(imgs) |
| 289 | seg_pred = seg_logit.argmax(dim=1) |
| 290 | seg_pred = seg_pred.cpu().numpy() |
| 291 | # unravel batch dim |
| 292 | seg_pred = list(seg_pred) |
| 293 | return seg_pred |