mrq/DL-Art-School
1
3
Fork 5
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix for randomresizedcrop injector

Browse Source
This commit is contained in:
James Betker 2021-01-07 16:31:43 -07:00
parent 659814c20f
commit acf1535b14
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
codes/trainer/injectors/base_injectors.py
View File

@ -379,9 +379,10 @@ class RandomCropInjector(Injector):
dim_in = opt['dim_in']
dim_out = opt['dim_out']
scale = dim_out / dim_in
self.operator = RandomResizedCrop(size=(dim_out, dim_out), scale=(scale, 1), ratio=(1,1),
self.operator = RandomResizedCrop(size=(dim_out, dim_out), scale=(scale, 1),
ratio=(.99,1), # An aspect ratio range is required, but .99,1 is effectively "none".
resample='NEAREST')
def forward(self, state):
return {self.output: self.operator(self.input)}
return {self.output: self.operator(state[self.input])}