mirror of
https://github.com/modelscope/DiffSynth-Studio.git
synced 2026-03-18 22:08:13 +00:00
Merge pull request #634 from modelscope/bugfix
fix videodataset to load images
This commit is contained in:
@@ -134,7 +134,8 @@ class VideoDataset(torch.utils.data.Dataset):
|
||||
def load_image(self, file_path):
|
||||
image = Image.open(file_path).convert("RGB")
|
||||
image = self.crop_and_resize(image, *self.get_height_width(image))
|
||||
return image
|
||||
frames = [image]
|
||||
return frames
|
||||
|
||||
|
||||
def is_image(self, file_path):
|
||||
|
||||
Reference in New Issue
Block a user