Spaces:
Configuration error
Configuration error
add timestamps in demo output
Browse files- demo.py +4 -0
- demo_dust3r_ga.py +2 -1
- 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
|
|
|
1 |
+
Subproject commit 9869e71f9165aa53c53ec0979cea1122a569ade4
|