Skip to content

Commit 39de31d

Browse files
committed
fix
1 parent 38d7719 commit 39de31d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

dataset.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -154,7 +154,7 @@ def __getitem__(self, index):
154154
or len(image_sequence) < self.sequence_length
155155
):
156156
image = Image.open(image_paths[i])
157-
image_tensor = self.transform(Image.open(image_paths[i]))
157+
image_tensor = self.transform(image)
158158
if flip:
159159
image_tensor = torch.flip(image_tensor, (-1,))
160160
image_sequence.append(image_tensor)

0 commit comments

Comments
 (0)