mirrors/mmsegmentation
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmsegmentation.git synced 2025-06-03 22:03:48 +08:00
Code Issues Projects Releases Wiki Activity

fix h,w order

Browse Source
This commit is contained in:
xiexinch 2022-12-01 11:25:00 +08:00
parent 721afaec31
commit 5a62b0335f
1 changed files with 1 additions and 1 deletions

2
mmseg/datasets/transforms/transforms.py

@ -1303,7 +1303,7 @@ class ResizeShortestEdge(BaseTransform):
new_h = int(new_h + 0.5) new_h = int(new_h + 0.5)
new_w = int(new_w + 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: def transform(self, results: Dict) -> Dict:
self.resize.scale = self._get_output_shape(results['img'], self.scale) self.resize.scale = self._get_output_shape(results['img'], self.scale)