Renamed out_folder for inference

mpelchat04 · Aug 6, 2020 · 99350bc · 99350bc
1 parent 4f99dc1
commit 99350bc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/airborne_lidar/airborne_lidar_inference.py b/examples/airborne_lidar/airborne_lidar_inference.py
@@ -110,7 +110,7 @@ def test(args, filename, info_class):
     print(f"Processing {filename}")
     las_filename = data_dir / f"{filename}.las"
     h5_filename = write_las_to_h5(Path(args['global']['data_dir']) / f"{filename}.las")
-    out_folder = model_folder.parent / 'tst'
+    out_folder = model_folder / 'tst'
     out_folder.mkdir(exist_ok=True)
 
     ds_tst = PartDatasetTest(h5_filename, folder=data_dir,  block_size=args['training']['blocksize'], npoints=args['training']['npoints'],