We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent f762d9f commit 74001a2Copy full SHA for 74001a2
utils/evaluation_utils.py
@@ -14,8 +14,9 @@ def read_samples(path):
14
15
tensor_list = []
16
for p in Path(path).rglob("*.png"):
17
- img = Image.open(p).convert("RGB")
18
- tensor_list.append(transform(img))
+ if "grid" not in str(p).split("/")[-1]:
+ img = Image.open(p).convert("RGB")
19
+ tensor_list.append(transform(img))
20
21
stacked_tensor = torch.stack(tensor_list, dim=0)
22
print(f"Read {len(stacked_tensor)} images")
0 commit comments