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 ac5c2b0 commit ee32b57Copy full SHA for ee32b57
applications/vision/data/imagenet/__init__.py
@@ -20,9 +20,6 @@ def make_data_reader(num_classes=1000, data_path=None):
20
print("Setting up data reader")
21
train_data_dir = os.path.join(data_path, 'train')
22
test_data_dir = os.path.join(data_path, 'val')
23
- #FIXME: This might not match the directory structure of the user's
24
- # dataset. We could infer the labels from the directory structure,
25
- # but the data reader is expecting a file.
26
train_label_file = os.path.join(data_path, 'labels/train.txt')
27
test_label_file = os.path.join(data_path, 'labels/val.txt')
28
0 commit comments