HuQiong
/
PaddleRS
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

permute->F.permute

Browse Source
own
Bobholamovic 2 years ago
parent 0962fb0ac3
commit e79d73faec
1 changed files with 9 additions and 9 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Unified View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 18
      paddlers/transforms/operators.py

18
paddlers/transforms/operators.py
Unescape View File

@ -1911,11 +1911,11 @@ class _Permute(Transform):
super(_Permute, self).__init__() super(_Permute, self).__init__()
def apply(self, sample): def apply(self, sample):
sample['image'] = permute(sample['image'], False) sample['image'] = F.permute(sample['image'], False)
if 'image2' in sample: if 'image2' in sample:
sample['image2'] = permute(sample['image2'], False) sample['image2'] = F.permute(sample['image2'], False)
if 'target' in sample: if 'target' in sample:
sample['target'] = permute(sample['target'], False) sample['target'] = F.permute(sample['target'], False)
return sample return sample
@ -2046,7 +2046,7 @@ class ArrangeSegmenter(Arrange):
mask = sample['mask'] mask = sample['mask']
mask = mask.astype('int64') mask = mask.astype('int64')
image = permute(sample['image'], False) image = F.permute(sample['image'], False)
if self.mode == 'train': if self.mode == 'train':
return image, mask return image, mask
if self.mode == 'eval': if self.mode == 'eval':
@ -2061,8 +2061,8 @@ class ArrangeChangeDetector(Arrange):
mask = sample['mask'] mask = sample['mask']
mask = mask.astype('int64') mask = mask.astype('int64')
image_t1 = permute(sample['image'], False) image_t1 = F.permute(sample['image'], False)
image_t2 = permute(sample['image2'], False) image_t2 = F.permute(sample['image2'], False)
if self.mode == 'train': if self.mode == 'train':
masks = [mask] masks = [mask]
if 'aux_masks' in sample: if 'aux_masks' in sample:
@ -2079,7 +2079,7 @@ class ArrangeChangeDetector(Arrange):
class ArrangeClassifier(Arrange): class ArrangeClassifier(Arrange):
def apply(self, sample): def apply(self, sample):
image = permute(sample['image'], False) image = F.permute(sample['image'], False)
if self.mode in ['train', 'eval']: if self.mode in ['train', 'eval']:
return image, sample['label'] return image, sample['label']
else: else:
@ -2096,8 +2096,8 @@ class ArrangeDetector(Arrange):
class ArrangeRestorer(Arrange): class ArrangeRestorer(Arrange):
def apply(self, sample): def apply(self, sample):
if 'target' in sample: if 'target' in sample:
target = permute(sample['target'], False) target = F.permute(sample['target'], False)
image = permute(sample['image'], False) image = F.permute(sample['image'], False)
if self.mode == 'train': if self.mode == 'train':
return image, target return image, target
if self.mode == 'eval': if self.mode == 'eval':

Write Preview
Loading…
Cancel
Save