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
numba # you should install numba==0.53.0 if your environment is cuda-9.0
numpy
nuscenes-devkit
open3d==0.16.0
open3d
plyfile
scikit-image
# by default we also use tensorboard to log results
......
# Copyright (c) OpenMMLab. All rights reserved.
import os
import os.path as osp
import tempfile
from unittest import TestCase
import mmengine
import numpy as np
import pytest
import torch
from mmengine.utils import is_list_of
......@@ -33,9 +35,11 @@ class TestLiDARSeg3DInferencer(TestCase):
np.allclose(pred1['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):
if not torch.cuda.is_available():
return
# single point cloud
inputs = dict(points='tests/data/s3dis/points/Area_1_office_2.bin')
torch.manual_seed(0)
......@@ -79,9 +83,11 @@ class TestLiDARSeg3DInferencer(TestCase):
self.assertIn('visualization', 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):
if not torch.cuda.is_available():
return
inputs = dict(points='tests/data/s3dis/points/Area_1_office_2.bin')
# img_out_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