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
951 stars 114 forks source link

VITON-HD inference needs not existing file #37

Closed jufilateva closed 2 months ago

jufilateva commented 3 months ago

Hello and thank you for your work!

I've tried to run an inference on VITON-HD that I've downloaded locally. And while inference I see this error:

Traceback (most recent call last): File "/home/cuda/CatVTON/inference.py", line 325, in main() File "/home/cuda/.local/lib/python3.10/site-packages/torch/utils/_contextlib.py", line 115, in decorate_context return func(*args, **kwargs) File "/home/cuda/CatVTON/inference.py", line 269, in main dataset = VITONHDTestDataset(args) File "/home/cuda/CatVTON/inference.py", line 18, in init self.data = self.load_data() File "/home/cuda/CatVTON/inference.py", line 39, in load_data assert os.path.exists(pair_txt:=os.path.join(self.args.data_root_path, 'test_pairs_unpaired.txt')), f"File {pair_txt} does not exist." AssertionError: File /home/cuda/zalando-hd-resized/test_pairs_unpaired.txt does not exist.

I've downloaded it again and rechecked file structure, but there is no test_pairs_unpaired.txt file in VITON-HD dataset. How could I avoud this problem?

Zheng-Chong commented 3 months ago

Please check /home/cuda/zalando-hd-resized/test_pairs_unpaired.txt in your local folder. Maybe your local .txt name is different, renamed it as test_pairs_unpaired.txt.

jufilateva commented 3 months ago

viton0hd test structure viton-hd structure

There are screens of dataset structure. Could the 'test_pairs.txt' be the file that sould be renamed?

Zheng-Chong commented 3 months ago

yes