-
Notifications
You must be signed in to change notification settings - Fork 0
/
generate_data.py
54 lines (35 loc) · 1.31 KB
/
generate_data.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
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
import cv2 as cv
import os
def pyramid_demo(image, s, index, dataset):
level = 3
temp = image.copy()
pyramid_images = []
for i in range(0, level):
dst = cv.pyrDown(temp)
pyramid_images.append(dst)
temp = dst.copy()
if i == 0:
cv.imwrite(dataset + '400_400/rgb/' + s[index], dst)
if i == 1:
cv.imwrite(dataset + '200_200/rgb/' + s[index], dst)
if i == 2:
cv.imwrite(dataset + '100_100/rgb/' + s[index], dst)
return pyramid_images
if __name__ == "__main__":
dataset = 'data/nsvf/Synthetic_NeRF/Chair'
path = dataset + '/rgb'
files = os.listdir(path)
s = []
for file_ in files:
if not os.path.isdir(path + file_):
f_name = str(file_)
s.append(f_name)
os.makedirs("{}100_100/rgb".format(dataset), exist_ok=False)
os.makedirs("{}200_200/rgb".format(dataset), exist_ok=False)
os.makedirs("{}400_400/rgb".format(dataset), exist_ok=False)
os.makedirs("{}800_800/rgb".format(dataset), exist_ok=False)
number = len(s)
for i in range(0, number):
src = cv.imread(dataset + '/rgb/' + s[i], -1)
cv.imwrite(dataset + '800_800/rgb/' + s[i], src)
pyramid_demo(src, s, i, dataset)