Simple test with single image.
(self, img, img_meta, rescale=True)
| 276 | return output |
| 277 | |
| 278 | def simple_test(self, img, img_meta, rescale=True): |
| 279 | """Simple test with single image.""" |
| 280 | seg_logit = self.inference(img, img_meta, rescale) |
| 281 | seg_pred = seg_logit.argmax(dim=1) |
| 282 | if torch.onnx.is_in_onnx_export(): |
| 283 | # our inference backend only support 4D output |
| 284 | seg_pred = seg_pred.unsqueeze(0) |
| 285 | return seg_pred |
| 286 | seg_pred = seg_pred.cpu().numpy() |
| 287 | # unravel batch dim |
| 288 | seg_pred = list(seg_pred) |
| 289 | return seg_pred |
| 290 | |
| 291 | def aug_test(self, imgs, img_metas, rescale=True): |
| 292 | """Test with augmentations. |