mirrors
/
mmyolo
mirror of https://github.com/open-mmlab/mmyolo.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Use Compose directly in BaseMixImageTransform (#71) 

* Use 'Compose' directly

Use 'Compose' directly

* avoids some unnecessary calculations

avoids some unnecessary calculations.

* remove whitespace

remove whitespace
pull/137/head
jason_w 2022-09-22 17:11:42 +08:00 committed by Haian Huang(深度眸)
parent 372f58759f
commit 41954a5ef5
1 changed files with 1 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
mmyolo/datasets/transforms/mix_img_transforms.py
View File

@ -42,11 +42,7 @@ class BaseMixImageTransform(BaseTransform, metaclass=ABCMeta):
if pre_transform is None:
self.pre_transform = None
else:
_transforms = []
for transform in pre_transform:
if isinstance(transform, dict):
_transforms.append(TRANSFORMS.build(transform))
self.pre_transform = Compose(_transforms)
self.pre_transform = Compose(pre_transform)
@abstractmethod
def get_indexes(self, dataset: BaseDataset) -> Union[list, int]: