423A35C7/pytorch3d
1
0
Fork 0
mirror of https://github.com/facebookresearch/pytorch3d.git synced 2025-12-20 06:10:34 +08:00
Code Issues Packages Projects Releases Wiki Activity

Optional ground-truth depth maps in visualiser

Browse Source 
Summary: The code does not crash if depth map/mask are not given.

Reviewed By: bottler

Differential Revision: D45082985

fbshipit-source-id: 3610d8beb4ac897fbbe52f56a6dd012a6365b89b
This commit is contained in:
Roman Shapovalov
2023-04-18 07:00:17 -07:00
committed by Facebook GitHub Bot
parent 1af6bf4768
commit 0e3138eca8
3 changed files with 42 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/implicitron/test_dataset_visualize.py
View File

@@ -26,6 +26,8 @@ from tests.common_testing import interactive_testing_requested
from .common_resources import get_skateboard_data
VISDOM_PORT = int(os.environ.get("VISDOM_PORT", 8097))
class TestDatasetVisualize(unittest.TestCase):
def setUp(self):
@@ -77,7 +79,7 @@ class TestDatasetVisualize(unittest.TestCase):
for k, dataset in self.datasets.items()
}
)
self.visdom = Visdom()
self.visdom = Visdom(port=VISDOM_PORT)
if not self.visdom.check_connection():
print("Visdom server not running! Disabling visdom visualizations.")
self.visdom = None