From 536aebd87425e7a6d56f15ac9b2c3b5318184c69 Mon Sep 17 00:00:00 2001 From: fe Date: Wed, 3 Jan 2024 12:36:50 -0800 Subject: [PATCH] Update __init__.py --- mask3d/__init__.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/mask3d/__init__.py b/mask3d/__init__.py index 9d153b3..358df36 100644 --- a/mask3d/__init__.py +++ b/mask3d/__init__.py @@ -29,7 +29,7 @@ def forward(self, x, raw_coordinates=None): import open3d as o3d # imports for output -from datasets.scannet200.scannet200_constants import (VALID_CLASS_IDS_20, VALID_CLASS_IDS_200, SCANNET_COLOR_MAP_20, SCANNET_COLOR_MAP_200) +from mask3d.datasets.scannet200.scannet200_constants import (VALID_CLASS_IDS_20, VALID_CLASS_IDS_200, SCANNET_COLOR_MAP_20, SCANNET_COLOR_MAP_200) def get_model(checkpoint_path=None): @@ -237,4 +237,4 @@ def save_colorized_mesh(mesh, labels_mapped, output_file, colormap='scannet'): # save colorized mesh save_colorized_mesh(mesh, labels, 'data/pcl_labelled.ply', colormap='scannet200') - \ No newline at end of file +