pytorch mobile trial
pytorch mobile trial with a self-trained gan(U-GAT-IT,pytorch version) , to make a person looks more beautiful
Training and test datasets are generated by styleGANv2
Effects:
Tips
0. not only girls ,but also boys works.
1. trained pytorch model should be converted via this method:
inut_tensor = torch.rand(1,3,256,256)
mobile = torch.jit.trace(processedModel,inut_tensor)
mobile.save("test2.pt")
processedModel is your model file and loaded by pytorch.