search

GrayXu / Virtual-Try-On-Flask

Virtual Try-on Network on Flask
Mozilla Public License 2.0
84 stars 20 forks source link

Run time error.output with shape [1, 256, 192] doesn't match the broadcast shape [3, 256, 192] #6

Open carfei opened 4 years ago

carfei commented 4 years ago

I also encountered the same problem when executing the program. I wonder if the author who had the same problem before has solved it? How to solve it? I really want to hear from you

farfun commented 3 years ago

transformer = transforms.Compose([transforms.ToTensor(), transforms.Normalize((0.5,0.5,0.5), (0.5,0.5,0.5)]) to transformer = transforms.Compose([transforms.ToTensor(), transforms.Normalize(0.5, 0.5)])