Skip to content

Commit 24d48e8

Browse files
author
Giancarlo
committed
corrected requirements
1 parent 95cc89a commit 24d48e8

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

requirements_demo.txt

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,4 +8,6 @@ Pillow==7.0.0
88
pyyaml
99
scikit-image
1010
imgaug
11-
sklearn
11+
sklearn
12+
tqdm
13+
natsort

test_fishy_torch.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,6 @@
66
import torchvision.transforms as transforms
77
from torchvision.transforms import ToPILImage, ToTensor
88
import yaml
9-
import bdlb
109
import random
1110
from options.test_options import TestOptions
1211
import sys
@@ -183,6 +182,7 @@ def estimator(image):
183182

184183

185184
if __name__ == '__main__':
185+
import bdlb
186186
# define fishyscapes test parameters
187187
fs = bdlb.load(benchmark="fishyscapes")
188188
# automatically downloads the dataset

0 commit comments

Comments
 (0)