15 lines
618 B
Python
15 lines
618 B
Python
|
def test_imagenet_data(imagenet):
|
||
|
|
total_length = len(imagenet)
|
||
|
|
assert total_length == 1281166 or total_length == 50000, 'The length of ImageNet is wrong : {}'.format(total_length)
|
||
|
|
map_id = {}
|
||
|
|
for index in range(total_length):
|
||
|
|
path, target = imagenet.imgs[index]
|
||
|
|
folder, image_name = os.path.split(path)
|
||
|
|
_, folder = os.path.split(folder)
|
||
|
|
if folder not in map_id:
|
||
|
|
map_id[folder] = target
|
||
|
|
else:
|
||
|
|
assert map_id[folder] == target, 'Class : {} is not {}'.format(folder, target)
|
||
|
|
assert image_name.find(folder) == 0, '{} is wrong.'.format(path)
|
||
|
|
print ('Check ImageNet Dataset OK')
|