Zheng-Chong / CatVTON

CatVTON is a simple and efficient virtual try-on diffusion model with 1) Lightweight Network (899.06M parameters totally), 2) Parameter-Efficient Training (49.57M parameters trainable) and 3) Simplified Inference (< 8G VRAM for 1024X768 resolution).
Other
964 stars 114 forks source link

FileNotFoundError: [Errno 2] No such file or directory: '/DressCode/upper_body/agnostic_masks/048402_0.png' Description: #32

Closed CrispyFeSo4 closed 3 months ago

CrispyFeSo4 commented 3 months ago

ERROR:FileNotFoundError: [Errno 2] No such file or directory: '/DressCode/upper_body/agnostic_masks/048402_0.png'

I found that the provided our preprocessed agnostic masks contains fewer than 1800 masks (upper/lower/dresses are all affected), which causes this error. How can this be resolved in such a case?

thanks!!

Zheng-Chong commented 3 months ago

Hi! I just wrote a script that allowed you to preprocess the agnostic masks by yourself, refer to https://github.com/zheng-chong/catvton?tab=readme-ov-file#1-data-preparation

CrispyFeSo4 commented 3 months ago

Issue resolved, thank you for your prompt and quick response! Good day!!