search

lastdefiance20 / TryYours-Virtual-Try-On

Virtual Try On site using HR-VITON. Anyone can try it using their own Image with clothes image.
102 stars 39 forks source link

Custom person Image #3

Closed Abdullahkhan5 closed 1 year ago

Abdullahkhan5 commented 1 year ago

Does this works with the custom data as well?

lastdefiance20 commented 1 year ago

If you were talking about using custom data that contains clothes and person images for Inference, I think it generates well even though there is some issue to solve like as background artifact.

you can always try your custom data at colab

stevemartinov commented 1 year ago

@lastdefiance20 Hi, given I have two input images (person and cloth), do you have instructions on how to run the code?

lastdefiance20 commented 1 year ago

@stevemartinov Hi, instructions are written at the colab. If you want to run the code locally, you need to use the docker image informed in "README.md", and follow all "Setup" / "Download Pre-trained Models" parts written at the colab before generating images.

After then, just replace the static/cloth_web.jpg file to your custom clothes .jpg file, and replace the static/origin_web.jpg file to your custom person .jpg file. You can generate the image running main.py. The generated image will be saved as static/finalimg.png

Abdullahkhan5 commented 1 year ago

Thanks for the help. I have successfully launched in on colab using my own custom dataset