From 6e682482d25e4df677fed6548c2f18c38eb3f097 Mon Sep 17 00:00:00 2001 From: Laurent FAINSIN Date: Tue, 16 May 2023 11:04:04 +0200 Subject: [PATCH] =?UTF-8?q?=F0=9F=90=9B=20missing=20imports?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- datasetss/S3DIS.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/datasetss/S3DIS.py b/datasetss/S3DIS.py index 83371ed..5270d3d 100644 --- a/datasetss/S3DIS.py +++ b/datasetss/S3DIS.py @@ -30,9 +30,9 @@ import torch import warnings from multiprocessing import Lock - +from utils.ply import write_ply, read_ply # OS functions -from os import listdir +from os import listdir, makedirs from os.path import exists, join, isdir # Dataset parent class @@ -757,6 +757,7 @@ class S3DISDataset(PointCloudDataset): cloud_classes = np.vstack((cloud_classes, object_classes)) # Save as ply + write_ply( cloud_file, (cloud_points, cloud_colors, cloud_classes), @@ -830,6 +831,8 @@ class S3DISDataset(PointCloudDataset): sub_labels = np.squeeze(sub_labels) # Get chosen neighborhoods + from sklearn.neighbors import KDTree + search_tree = KDTree(sub_points, leaf_size=10) # search_tree = nnfln.KDTree(n_neighbors=1, metric='L2', leaf_size=10) # search_tree.fit(sub_points)