-
Notifications
You must be signed in to change notification settings - Fork 0
/
generate_gallery_nerf.py
executable file
·51 lines (40 loc) · 1.8 KB
/
generate_gallery_nerf.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
import os
template = """
<div class="col-md-6 captioned_videos" id="res_{vid_id}" style="margin-top: 0px; margin-bottom: 0px ;">
<video class="video lazy" loop playsinline autoplay muted style="width: 100%">
<source data-src="./static/NeRF-Dreamfusion/{keyword}.mp4" type="video/mp4"></source>
</video>
<p class="compare-caption">{description}</p>
</div>"""
video_tokens = []
keys = []
with open("./static/nerf-dreamfusion.txt", "r") as f:
for line in f:
video_names = line.strip()
video_tokens.append(video_names)
keys.append(os.path.splitext(os.path.basename(video_names))[0])
with open("gallery_template.html", "r") as f:
html_template = f.read()
content = ""
page_idx = 0
prev_i = 0
page_size = 10
for i,k in enumerate(keys):
description = keys[i]
current_video = template.format(vid_id=i, keyword=k, description=description)
content += current_video
first_page = page_idx - 1 < 0
last_page = i + 1 >= len(keys)
if i - prev_i + 1 >= page_size or last_page:
link1 = f"nerf_dreamfusion_gallery_{page_idx - 1}.html"
link2 = f"nerf_dreamfusion_gallery_{page_idx + 1}.html"
active1 = "disabled" if first_page else ""
active2 = "disabled" if last_page else ""
html = html_template.replace("{{{content}}}", content).replace("{{{link1}}}", link1).replace("{{{link2}}}", link2)
html = html.replace("{{{active1}}}", active1).replace("{{{active2}}}", active2)
html = html.replace("{{{start_idx}}}", str(prev_i+1)).replace("{{{end_idx}}}", str(i+1)).replace("{{{total}}}", str(len(keys)))
with open(f"nerf_dreamfusion_gallery_{page_idx}.html", "w") as f:
f.write(html)
content = ""
page_idx += 1
prev_i = i + 1