Skip to content
Snippets Groups Projects
Commit 696952ff authored by Andrei-Claudiu Roibu's avatar Andrei-Claudiu Roibu :desktop:
Browse files

added scales to preseve the object dimensions

parent e342fa1f
No related branches found
No related merge requests found
......@@ -305,9 +305,9 @@ class AddSingleTransformation(object):
elif self.transformationFlag == 'RandomAffine':
if self.fix_seed == True:
torch.manual_seed(0)
self.transformation = tio.RandomAffine(degrees=self.transformationMetric)
self.transformation = tio.RandomAffine(scales = 0, degrees=self.transformationMetric)
else:
self.transformation = tio.RandomAffine(degrees=self.transformationMetric)
self.transformation = tio.RandomAffine(scales = 0, degrees=self.transformationMetric)
elif self.transformationFlag == 'RandomNoiseMask':
if self.fix_seed == True:
torch.manual_seed(0)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment