GrayXu / Virtual-Try-On-Flask

Virtual Try-on Network on Flask

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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

saisusrith opened this issue · comments

Hi while running the your template.ipynb we get the following error.
Even after debugging I could not find the error.
RuntimeError: output with shape [1, 256, 192] doesn't match the broadcast shape [3, 256, 192]

Hi, thanks for this issue.

Can you copy complete error logs, like which line incurred this error?

Hi GrayXu,we debugged your code . The error is at line 179 "one_map = Model.transformer(one_map)".After it calls the Model.transformer error pops up

Hi @GrayXu any solution?

Sry,
because of Coronavirus vacation in China, I can't access GPU machines to test codes these days.
I don't find errors in Model.transformer. The transformer' just tensorlize and normalize data.
Can you show your local enviroments? I only test it on torch==1.2.0 torchvision==0.2.0 with GPU.

I also encountered the same error message, don't know if you solved it? How to solve it?

@GrayXu ,An error occurred in getPoseMap(self, pose_data),The number of output channels changed from 3 to 1.So how to solve it?

Sry,
because of Coronavirus vacation in China, I can't access GPU machines to test codes these days.
I don't find errors in Model.transformer. The transformer' just tensorlize and normalize data.
Can you show your local enviroments? I only test it on torch==1.2.0 torchvision==0.2.0 with GPU.

i use torch==1.2.0 torchvision==0.2.0 with GPU, but the error still exists.

Sry,
because of Coronavirus vacation in China, I can't access GPU machines to test codes these days.
I don't find errors in Model.transformer. The transformer' just tensorlize and normalize data.
Can you show your local enviroments? I only test it on torch==1.2.0 torchvision==0.2.0 with GPU.

i use torch==1.2.0 torchvision==0.2.0 with GPU, but the error still exists.

solved. input image size should be 192*256.

Hi @mad-fogs ,
When I am clicking the button for selecting the cloth, it throws an error on submitting. We are saving them in static/img folder right?