Spaces:
Runtime error
Runtime error
update: diable gr.info notice
Browse files- common/utils.py +7 -7
common/utils.py
CHANGED
|
@@ -463,7 +463,6 @@ def run_ransac(
|
|
| 463 |
ransac_confidence=ransac_confidence,
|
| 464 |
ransac_max_iter=ransac_max_iter,
|
| 465 |
)
|
| 466 |
-
gr.Info(f"RANSAC matches done using: {time.time()-t1:.3f}s")
|
| 467 |
logger.info(f"RANSAC matches done using: {time.time()-t1:.3f}s")
|
| 468 |
t1 = time.time()
|
| 469 |
|
|
@@ -475,7 +474,6 @@ def run_ransac(
|
|
| 475 |
output_matches_ransac, num_matches_ransac = display_matches(
|
| 476 |
state_cache, titles=titles, tag="KPTS_RANSAC"
|
| 477 |
)
|
| 478 |
-
gr.Info(f"Display matches done using: {time.time()-t1:.3f}s")
|
| 479 |
logger.info(f"Display matches done using: {time.time()-t1:.3f}s")
|
| 480 |
t1 = time.time()
|
| 481 |
num_matches_raw = state_cache["num_matches_raw"]
|
|
@@ -561,7 +559,7 @@ def run_matching(
|
|
| 561 |
else:
|
| 562 |
matcher = get_model(match_conf)
|
| 563 |
models_already_loaded[cache_key] = matcher
|
| 564 |
-
gr.Info(f"Loading model using: {time.time()-t0:.3f}s")
|
| 565 |
logger.info(f"Loading model using: {time.time()-t0:.3f}s")
|
| 566 |
t1 = time.time()
|
| 567 |
|
|
@@ -593,7 +591,9 @@ def run_matching(
|
|
| 593 |
)
|
| 594 |
pred = match_features.match_images(matcher, pred0, pred1)
|
| 595 |
del extractor
|
| 596 |
-
gr.Info(
|
|
|
|
|
|
|
| 597 |
logger.info(f"Matching images done using: {time.time()-t1:.3f}s")
|
| 598 |
t1 = time.time()
|
| 599 |
# plot images with keypoints\
|
|
@@ -627,7 +627,7 @@ def run_matching(
|
|
| 627 |
ransac_confidence=ransac_confidence,
|
| 628 |
ransac_max_iter=ransac_max_iter,
|
| 629 |
)
|
| 630 |
-
gr.Info(f"RANSAC matches done using: {time.time()-t1:.3f}s")
|
| 631 |
logger.info(f"RANSAC matches done using: {time.time()-t1:.3f}s")
|
| 632 |
t1 = time.time()
|
| 633 |
|
|
@@ -639,7 +639,7 @@ def run_matching(
|
|
| 639 |
output_matches_ransac, num_matches_ransac = display_matches(
|
| 640 |
pred, titles=titles, tag="KPTS_RANSAC"
|
| 641 |
)
|
| 642 |
-
gr.Info(f"Display matches done using: {time.time()-t1:.3f}s")
|
| 643 |
logger.info(f"Display matches done using: {time.time()-t1:.3f}s")
|
| 644 |
|
| 645 |
t1 = time.time()
|
|
@@ -653,8 +653,8 @@ def run_matching(
|
|
| 653 |
)
|
| 654 |
plt.close("all")
|
| 655 |
# del pred
|
|
|
|
| 656 |
logger.info(f"TOTAL time: {time.time()-t0:.3f}s")
|
| 657 |
-
gr.Info(f"In summary, total time: {time.time()-t0:.3f}s")
|
| 658 |
|
| 659 |
state_cache = pred
|
| 660 |
state_cache["num_matches_raw"] = num_matches_raw
|
|
|
|
| 463 |
ransac_confidence=ransac_confidence,
|
| 464 |
ransac_max_iter=ransac_max_iter,
|
| 465 |
)
|
|
|
|
| 466 |
logger.info(f"RANSAC matches done using: {time.time()-t1:.3f}s")
|
| 467 |
t1 = time.time()
|
| 468 |
|
|
|
|
| 474 |
output_matches_ransac, num_matches_ransac = display_matches(
|
| 475 |
state_cache, titles=titles, tag="KPTS_RANSAC"
|
| 476 |
)
|
|
|
|
| 477 |
logger.info(f"Display matches done using: {time.time()-t1:.3f}s")
|
| 478 |
t1 = time.time()
|
| 479 |
num_matches_raw = state_cache["num_matches_raw"]
|
|
|
|
| 559 |
else:
|
| 560 |
matcher = get_model(match_conf)
|
| 561 |
models_already_loaded[cache_key] = matcher
|
| 562 |
+
# gr.Info(f"Loading model using: {time.time()-t0:.3f}s")
|
| 563 |
logger.info(f"Loading model using: {time.time()-t0:.3f}s")
|
| 564 |
t1 = time.time()
|
| 565 |
|
|
|
|
| 591 |
)
|
| 592 |
pred = match_features.match_images(matcher, pred0, pred1)
|
| 593 |
del extractor
|
| 594 |
+
gr.Info(
|
| 595 |
+
f"Matching images done using: {time.time()-t1:.3f}s",
|
| 596 |
+
)
|
| 597 |
logger.info(f"Matching images done using: {time.time()-t1:.3f}s")
|
| 598 |
t1 = time.time()
|
| 599 |
# plot images with keypoints\
|
|
|
|
| 627 |
ransac_confidence=ransac_confidence,
|
| 628 |
ransac_max_iter=ransac_max_iter,
|
| 629 |
)
|
| 630 |
+
# gr.Info(f"RANSAC matches done using: {time.time()-t1:.3f}s")
|
| 631 |
logger.info(f"RANSAC matches done using: {time.time()-t1:.3f}s")
|
| 632 |
t1 = time.time()
|
| 633 |
|
|
|
|
| 639 |
output_matches_ransac, num_matches_ransac = display_matches(
|
| 640 |
pred, titles=titles, tag="KPTS_RANSAC"
|
| 641 |
)
|
| 642 |
+
# gr.Info(f"Display matches done using: {time.time()-t1:.3f}s")
|
| 643 |
logger.info(f"Display matches done using: {time.time()-t1:.3f}s")
|
| 644 |
|
| 645 |
t1 = time.time()
|
|
|
|
| 653 |
)
|
| 654 |
plt.close("all")
|
| 655 |
# del pred
|
| 656 |
+
# gr.Info(f"In summary, total time: {time.time()-t0:.3f}s")
|
| 657 |
logger.info(f"TOTAL time: {time.time()-t0:.3f}s")
|
|
|
|
| 658 |
|
| 659 |
state_cache = pred
|
| 660 |
state_cache["num_matches_raw"] = num_matches_raw
|