yocabon commited on
Commit
427435d
·
1 Parent(s): d888156

add timestamps in demo output

Browse files
Files changed (3) hide show
  1. demo.py +4 -0
  2. demo_dust3r_ga.py +2 -1
  3. dust3r +1 -1
demo.py CHANGED
@@ -15,6 +15,9 @@ from mast3r.demo import get_args_parser, main_demo
15
  from mast3r.model import AsymmetricMASt3R
16
  from mast3r.utils.misc import hash_md5
17
 
 
 
 
18
  import matplotlib.pyplot as pl
19
  pl.ion()
20
 
@@ -23,6 +26,7 @@ torch.backends.cuda.matmul.allow_tf32 = True # for gpu >= Ampere and pytorch >=
23
  if __name__ == '__main__':
24
  parser = get_args_parser()
25
  args = parser.parse_args()
 
26
 
27
  if args.server_name is not None:
28
  server_name = args.server_name
 
15
  from mast3r.model import AsymmetricMASt3R
16
  from mast3r.utils.misc import hash_md5
17
 
18
+ import mast3r.utils.path_to_dust3r # noqa
19
+ from dust3r.demo import set_print_with_timestamp
20
+
21
  import matplotlib.pyplot as pl
22
  pl.ion()
23
 
 
26
  if __name__ == '__main__':
27
  parser = get_args_parser()
28
  args = parser.parse_args()
29
+ set_print_with_timestamp()
30
 
31
  if args.server_name is not None:
32
  server_name = args.server_name
demo_dust3r_ga.py CHANGED
@@ -13,7 +13,7 @@ import mast3r.utils.path_to_dust3r # noqa
13
  from dust3r.model import AsymmetricCroCo3DStereo
14
  from mast3r.model import AsymmetricMASt3R
15
  from dust3r.demo import get_args_parser as dust3r_get_args_parser
16
- from dust3r.demo import main_demo
17
 
18
  import matplotlib.pyplot as pl
19
  pl.ion()
@@ -36,6 +36,7 @@ def get_args_parser():
36
  if __name__ == '__main__':
37
  parser = get_args_parser()
38
  args = parser.parse_args()
 
39
 
40
  if args.tmp_dir is not None:
41
  tmp_path = args.tmp_dir
 
13
  from dust3r.model import AsymmetricCroCo3DStereo
14
  from mast3r.model import AsymmetricMASt3R
15
  from dust3r.demo import get_args_parser as dust3r_get_args_parser
16
+ from dust3r.demo import main_demo, set_print_with_timestamp
17
 
18
  import matplotlib.pyplot as pl
19
  pl.ion()
 
36
  if __name__ == '__main__':
37
  parser = get_args_parser()
38
  args = parser.parse_args()
39
+ set_print_with_timestamp()
40
 
41
  if args.tmp_dir is not None:
42
  tmp_path = args.tmp_dir
dust3r CHANGED
@@ -1 +1 @@
1
- Subproject commit bac527a8daa1e38a7de58bfa12aac4d3bf472dd0
 
1
+ Subproject commit 9869e71f9165aa53c53ec0979cea1122a569ade4