miccunifi / ladi-vton

[ACM MM 2023] - LaDI-VTON: Latent Diffusion Textual-Inversion Enhanced Virtual Try-On
Other
412 stars 56 forks source link

Batch images not being processed by AutoEncoder #53

Open Aukture opened 5 months ago

Aukture commented 5 months ago
  File "D:\RK\Projects\FashionHelp\Virtual-TryOn\fashionAI\FashionAI\vto_stablediffusion\lib\ladi-vton\src\inference.py", line 285, in main
    processed_images = processor(images=input_image, return_tensors="pt")
  File "C:\CustomInstalls\anaconda3\envs\ladi-vton\lib\site-packages\transformers\models\clip\processing_clip.py", line 85, in __call__
    image_features = self.feature_extractor(images, return_tensors=return_tensors, **kwargs)
  File "C:\CustomInstalls\anaconda3\envs\ladi-vton\lib\site-packages\transformers\image_processing_utils.py", line 437, in __call__
    return self.preprocess(images, **kwargs)
  File "C:\CustomInstalls\anaconda3\envs\ladi-vton\lib\site-packages\transformers\models\clip\image_processing_clip.py", line 331, in preprocess
    images = [self.resize(image=image, size=size, resample=resample) for image in images]
  File "C:\CustomInstalls\anaconda3\envs\ladi-vton\lib\site-packages\transformers\models\clip\image_processing_clip.py", line 331, in <listcomp>
    images = [self.resize(image=image, size=size, resample=resample) for image in images]
  File "C:\CustomInstalls\anaconda3\envs\ladi-vton\lib\site-packages\transformers\models\clip\image_processing_clip.py", line 157, in resize
    return resize(image, size=output_size, resample=resample, data_format=data_format, **kwargs)
  File "C:\CustomInstalls\anaconda3\envs\ladi-vton\lib\site-packages\transformers\image_transforms.py", line 265, in resize
    image = to_channel_dimension_format(image, ChannelDimension.LAST)
  File "C:\CustomInstalls\anaconda3\envs\ladi-vton\lib\site-packages\transformers\image_transforms.py", line 75, in to_channel_dimension_format
    image = image.transpose((1, 2, 0))