Add OPC wrapper

SuperBuild/cmake/External-OpenPointClass.cmake

@@ -17,6 +17,7 @@ ExternalProject_Add(${_proj_name}
   CMAKE_ARGS
     -DPDAL_DIR=${SB_INSTALL_DIR}/lib/cmake/PDAL
     -DWITH_GBT=ON
+    -DBUILD_PCTRAIN=OFF
     -DEIGEN3_INCLUDE_DIR=${SB_SOURCE_DIR}/eigen34/
     -DCMAKE_BUILD_TYPE=${CMAKE_BUILD_TYPE}
     -DCMAKE_INSTALL_PREFIX:PATH=${SB_INSTALL_DIR}

opendm/opc.py

@@ -0,0 +1,29 @@
+import os
+from opendm.ai import get_model
+from opendm import log
+from opendm.system import run
+from opendm import io
+
+def classify(point_cloud):
+    tmp_output = io.related_file_path(point_cloud, postfix=".classified")
+    if os.path.isfile(tmp_output):
+        os.remove(tmp_output)
+
+    try:
+        model = get_model("openpointclass", 
+            "https://github.com/uav4geo/OpenPointClass/releases/download/v1.1.3/vehicles-vegetation-buildings.zip", 
+            "v1.0.0",
+            name="model.bin")
+
+        if model is not None:
+            run('pcclassify "%s" "%s" "%s" -u -s 2,64' % (point_cloud, tmp_output, model))
+        else:
+            log.ODM_WARNING("Cannot download/access model from %s" % (model_url))
+    except Exception as e:
+        log.ODM_WARNING("Cannot classify using OpenPointClass: %s" % str(e))
+
+    if os.path.isfile(tmp_output):
+        os.remove(point_cloud)
+        os.rename(tmp_output, point_cloud)
+    else:
+        log.ODM_WARNING("Cannot classify using OpenPointClass (no output generated)")

stages/odm_dem.py

@@ -12,7 +12,7 @@ from opendm.cropper import Cropper
 from opendm import pseudogeo
 from opendm.tiles.tiler import generate_dem_tiles
 from opendm.cogeo import convert_to_cogeo
-
+from opendm.opc import classify
 
 class ODMDEMStage(types.ODM_Stage):
     def process(self, args, outputs):
@@ -49,7 +49,7 @@ class ODMDEMStage(types.ODM_Stage):
             pc_classify_marker = os.path.join(odm_dem_root, 'pc_classify_done.txt')
 
             if not io.file_exists(pc_classify_marker) or self.rerun():
-                log.ODM_INFO("Classifying {} using Simple Morphological Filter".format(dem_input))
+                log.ODM_INFO("Classifying {} using Simple Morphological Filter (1/2)".format(dem_input))
                 commands.classify(dem_input,
                                   args.smrf_scalar, 
                                   args.smrf_slope, 
@@ -57,6 +57,9 @@ class ODMDEMStage(types.ODM_Stage):
                                   args.smrf_window
                                 )
 
+                log.ODM_INFO("Classifying {} using OpenPointClass (2/2)".format(dem_input))
+                classify(dem_input)
+
                 with open(pc_classify_marker, 'w') as f:
                     f.write('Classify: smrf\n')
                     f.write('Scalar: {}\n'.format(args.smrf_scalar))