github/ai-toolkit
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Fixed issues with vgg19 preprocessing. Added yaml support on config file

Browse Source
This commit is contained in:
Jaret Burkett
2023-07-19 06:12:57 -06:00
parent 8d8229dfc0
commit 6ada328d8d
6 changed files with 46 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
toolkit/style.py
View File

@@ -121,8 +121,9 @@ class Normalization(nn.Module):
# normalize to min and max of 0 - 1
in_min = torch.min(stacked_input)
in_max = torch.max(stacked_input)
norm_stacked_input = (stacked_input - in_min) / (in_max - in_min)
return (norm_stacked_input - self.mean) / self.std
# norm_stacked_input = (stacked_input - in_min) / (in_max - in_min)
# return (norm_stacked_input - self.mean) / self.std
return (stacked_input - self.mean) / self.std
def get_style_model_and_losses(