-
Notifications
You must be signed in to change notification settings - Fork 3
/
make_SUNRGBD.py
36 lines (27 loc) · 1.11 KB
/
make_SUNRGBD.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
import os
import json
from pathlib import Path
from scipy.misc import imread
train_13_txt = Path('./data/SUNRGBD_train13.txt')
test_13_txt = Path('./data/SUNRGBD_test13.txt')
train_37_txt = Path('./data/SUNRGBD_train37.txt')
test_37_txt = Path('./data/SUNRGBD_test37.txt')
train_13_odgt = Path('./data/train_13_SUNRGBD.odgt')
test_13_odgt = Path('./data/test_13_SUNRGBD.odgt')
train_37_odgt = Path('./data/train_37_SUNRGBD.odgt')
test_37_odgt = Path('./data/test_37_SUNRGBD.odgt')
def make_SUNRGBD(txt, odgt):
with txt.open(mode='r') as fi:
with odgt.open(mode='w ') as fo:
lines = [line.rstrip('\n').split(' ') for line in fi]
for l in lines:
img = imread('./data/' l[0], mode='RGB')
item = {"width": img.shape[1], "fpath_img": l[0], "height": img.shape[0], "fpath_segm": l[2]}
fo.write(f'{json.dumps(item)}\n')
def main():
make_SUNRGBD(train_13_txt, train_13_odgt)
make_SUNRGBD(test_13_txt, test_13_odgt)
make_SUNRGBD(train_37_txt, train_37_odgt)
make_SUNRGBD(test_37_txt, test_37_odgt)
if __name__ == '__main__':
main()