Skip to content

Commit

Permalink
Use all five vgg
Browse files Browse the repository at this point in the history
  • Loading branch information
SunshineAtNoon authored Jul 24, 2018
1 parent dad6c83 commit 4a3d674
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion WCT.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -56,7 +56,7 @@ def styleTransfer(contentImg,styleImg,imname,csF):
Im5 = wct.d5(csF5)

sF4 = wct.e4(styleImg)
cF4 = wct.e4(contentImg)
cF4 = wct.e4(Im5)
sF4 = sF4.data.cpu().squeeze(0)
cF4 = cF4.data.cpu().squeeze(0)
csF4 = wct.transform(cF4,sF4,csF,args.alpha)
Expand Down

0 comments on commit 4a3d674

Please sign in to comment.