fix h,w order

2022-12-01 11:25:00 +08:00 · 2022-12-01 11:25:00 +08:00 · 5a62b0335f
parent 721afaec31
commit 5a62b0335f
1 changed files with 1 additions and 1 deletions
--- a/mmseg/datasets/transforms/transforms.py
+++ b/mmseg/datasets/transforms/transforms.py
@ -1303,7 +1303,7 @@ class ResizeShortestEdge(BaseTransform):

        new_h = int(new_h + 0.5)
        new_w = int(new_w + 0.5)
-        return (new_h, new_w)
+        return (new_w, new_h)

    def transform(self, results: Dict) -> Dict:
        self.resize.scale = self._get_output_shape(results['img'], self.scale)