Simple test with single image.
(self, img, img_meta, rescale=True)
| 261 | return output |
| 262 | |
| 263 | def simple_test(self, img, img_meta, rescale=True): |
| 264 | """Simple test with single image.""" |
| 265 | seg_logit = self.inference(img, img_meta, rescale) |
| 266 | seg_pred = seg_logit.argmax(dim=1) |
| 267 | if torch.onnx.is_in_onnx_export(): |
| 268 | # our inference backend only support 4D output |
| 269 | seg_pred = seg_pred.unsqueeze(0) |
| 270 | return seg_pred |
| 271 | seg_pred = seg_pred.cpu().numpy() |
| 272 | # unravel batch dim |
| 273 | seg_pred = list(seg_pred) |
| 274 | return seg_pred |
| 275 | |
| 276 | def aug_test(self, imgs, img_metas, rescale=True): |
| 277 | """Test with augmentations. |