verify_objects365.py 531 Bytes
Newer Older
chenzk's avatar
v1.0  
chenzk committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
import os
from tqdm import tqdm

dir = '../datasets/Objects365v1/labels/train'

count = 0
corrupt_count = 0
for file in tqdm(os.listdir(dir)):
    with open(f'{dir}/{file}', 'r') as f:
        lines = f.readlines()
    assert(len(lines) == len(set(lines)))
    count += len(lines)
    for line in lines:
        line = line.strip()
        line = line.split()
        line = [float(l) for l in line[1:]]
        assert(all([l <= 1 for l in line]))
        assert(all([l >= 0 for l in line]))

assert(count == 8529995)