theluyuan/DiffSynth-Studio
1
0
Fork 0
mirror of https://github.com/modelscope/DiffSynth-Studio.git synced 2026-03-24 10:18:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix bugs

Browse Source
This commit is contained in:
wang96
2025-02-27 15:26:39 +08:00
parent 0aca943a39
commit 490d420d82
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
examples/wanvideo/train_wan_i2v.py
View File

@@ -24,8 +24,8 @@ class I2VDataset(torch.utils.data.Dataset):
self.width = width self.width = width
self.frame_process = v2.Compose([ self.frame_process = v2.Compose([
v2.Resize(size=(height, width), antialias=True),
v2.CenterCrop(size=(height, width)), v2.CenterCrop(size=(height, width)),
v2.Resize(size=(height, width), antialias=True),
v2.ToTensor(), v2.ToTensor(),
v2.Normalize(mean=[0.5, 0.5, 0.5], std=[0.5, 0.5, 0.5]), v2.Normalize(mean=[0.5, 0.5, 0.5], std=[0.5, 0.5, 0.5]),
]) ])
@@ -80,7 +80,7 @@ class I2VDataset(torch.utils.data.Dataset):
def __getitem__(self, data_id): def __getitem__(self, data_id):
text = self.path[data_id] text = self.text[data_id]
path = self.path[data_id] path = self.path[data_id]
video, first_frame_image = self.load_video(path) video, first_frame_image = self.load_video(path)
data = {"text": text, "video": video, "first_frame_image":np.array(first_frame_image), "path": path} data = {"text": text, "video": video, "first_frame_image":np.array(first_frame_image), "path": path}