Unverified Commit 5cd59f78 authored by Sun Jiahao's avatar Sun Jiahao Committed by GitHub
Browse files

[Fix]: fix open3d version bug (#2633)

parent 5d822daa
...@@ -3,7 +3,7 @@ networkx>=2.5 ...@@ -3,7 +3,7 @@ networkx>=2.5
numba # you should install numba==0.53.0 if your environment is cuda-9.0 numba # you should install numba==0.53.0 if your environment is cuda-9.0
numpy numpy
nuscenes-devkit nuscenes-devkit
open3d==0.16.0 open3d
plyfile plyfile
scikit-image scikit-image
# by default we also use tensorboard to log results # by default we also use tensorboard to log results
......
# Copyright (c) OpenMMLab. All rights reserved. # Copyright (c) OpenMMLab. All rights reserved.
import os
import os.path as osp import os.path as osp
import tempfile import tempfile
from unittest import TestCase from unittest import TestCase
import mmengine import mmengine
import numpy as np import numpy as np
import pytest
import torch import torch
from mmengine.utils import is_list_of from mmengine.utils import is_list_of
...@@ -33,9 +35,11 @@ class TestLiDARSeg3DInferencer(TestCase): ...@@ -33,9 +35,11 @@ class TestLiDARSeg3DInferencer(TestCase):
np.allclose(pred1['pts_semantic_mask'], np.allclose(pred1['pts_semantic_mask'],
pred2['pts_semantic_mask'])) pred2['pts_semantic_mask']))
@pytest.mark.skipif(
not torch.cuda.is_available(), reason='requires CUDA support')
@pytest.mark.skipif(
'DISPLAY' not in os.environ, reason='requires DISPLAY device')
def test_call(self): def test_call(self):
if not torch.cuda.is_available():
return
# single point cloud # single point cloud
inputs = dict(points='tests/data/s3dis/points/Area_1_office_2.bin') inputs = dict(points='tests/data/s3dis/points/Area_1_office_2.bin')
torch.manual_seed(0) torch.manual_seed(0)
...@@ -79,9 +83,11 @@ class TestLiDARSeg3DInferencer(TestCase): ...@@ -79,9 +83,11 @@ class TestLiDARSeg3DInferencer(TestCase):
self.assertIn('visualization', res_bs2) self.assertIn('visualization', res_bs2)
self.assertIn('predictions', res_bs2) self.assertIn('predictions', res_bs2)
@pytest.mark.skipif(
not torch.cuda.is_available(), reason='requires CUDA support')
@pytest.mark.skipif(
'DISPLAY' not in os.environ, reason='requires DISPLAY device')
def test_visualizer(self): def test_visualizer(self):
if not torch.cuda.is_available():
return
inputs = dict(points='tests/data/s3dis/points/Area_1_office_2.bin') inputs = dict(points='tests/data/s3dis/points/Area_1_office_2.bin')
# img_out_dir # img_out_dir
with tempfile.TemporaryDirectory() as tmp_dir: with tempfile.TemporaryDirectory() as tmp_dir:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment