search

yisol / IDM-VTON

[ECCV2024] IDM-VTON : Improving Diffusion Models for Authentic Virtual Try-on in the Wild
https://idm-vton.github.io/
3.68k stars 583 forks source link

How can I integrate my own masking models? YoloV8 #58

Closed dancingninjaaaaa closed 4 months ago

dancingninjaaaaa commented 4 months ago

Hi @yisol, I want to change your pipeline a bit, I want to integrate my own, masking models. I want to integrate yolov8 lower body, and full body inside it. Can you someone please provide some guidance.

Manish989927 commented 4 months ago

I think use this https://colab.research.google.com/drive/1aMbKHn5dVXwhnTtC9NIWm3KDG7bzNjrF?usp=sharing that help you to understand

dancingninjaaaaa commented 4 months ago

Hi, not this, I already have trained yolov8, I just want to integrate them inside IDM-VTON.

dancingninjaaaaa commented 4 months ago

@yisol From where you are importing openposedetector? Can't find it...