From 61801681178673bb11a9f244ffb67a8d1b2af186 Mon Sep 17 00:00:00 2001 From: Mathieu Turgeon-Pelchat Date: Tue, 2 Nov 2021 11:53:40 -0400 Subject: [PATCH] bug fix --- airborne_lidar/airborne_lidar_seg.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/airborne_lidar/airborne_lidar_seg.py b/airborne_lidar/airborne_lidar_seg.py index cb96d27..963d5ae 100644 --- a/airborne_lidar/airborne_lidar_seg.py +++ b/airborne_lidar/airborne_lidar_seg.py @@ -17,11 +17,10 @@ import torch.nn.functional as F import convpoint.knn.lib.python.nearest_neighbors as nearest_neighbors import utils.metrics as metrics -from airborne_lidar.airborne_lidar_utils import InformationLogger, print_metric, write_config, read_parameters, wblue, wgreen, write_features +from airborne_lidar.airborne_lidar_utils import InformationLogger, print_metric, write_config, read_parameters, wblue, wgreen, write_features, write_to_las from pathlib import Path from airborne_lidar.airborne_lidar_datasets import PartDatasetTrainVal, PartDatasetTest from airborne_lidar.prepare_airborne_lidar_label import read_las_format -from airborne_lidar.airborne_lidar_inference import write_to_las from mlflow import log_params, set_tracking_uri, set_experiment