def forward(self, image: torch.Tensor) -> torch.Tensor:
    img = self.preprocess(image)
    out = self.model(img)
    return self.postprocess(out, img.shape[(- 2)], img.shape[(- 1)])
def forward(self, image: torch.Tensor) -> torch.Tensor:
    img = self.preprocess(image)
    out = self.model(img)
    return self.postprocess(out, img.shape[(- 2)], img.shape[(- 1)])