-
Notifications
You must be signed in to change notification settings - Fork 1
/
display.py
44 lines (31 loc) · 998 Bytes
/
display.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
print("Loading the required Python modules for the ResNet152 model ...")
from objreg_utils import (
_save_tf_model_graph,
_MODEL_FILE,
)
import shutil
import subprocess
import signal
import time
print("\nLoading the pre-trained ResNet v1 152 model ...")
logdir = _save_tf_model_graph(_MODEL_FILE)
# Run TensorBoard for visualization
print("\nStarting TensorBoard ...")
tb_proc = subprocess.Popen(
"exec tensorboard --logdir={}".format(logdir),
shell=True,
stderr=subprocess.PIPE)
# Wait 1 secs for tensorboard fully started
print("\nOpening browser for visualizing the model graph ..."
"\n (Please refresh the browser if fail to connect localhost:6006.)")
time.sleep(2)
br_proc = subprocess.Popen(
"xdg-open http://localhost:6006",
shell=True,
stderr=subprocess.PIPE)
# Wait for browser to be closed
output, errors = br_proc.communicate()
# Send Ctrl-C to tensorboard
tb_proc.send_signal(signal.SIGINT)
# Remove log files
shutil.rmtree(logdir)