Respect max-concurrency when calling OPC

pull/1631/head
Piero Toffanin 2023-03-20 23:49:23 -04:00
rodzic 7cbe959da6
commit 976db04148
2 zmienionych plików z 3 dodań i 3 usunięć

Wyświetl plik

@ -4,7 +4,7 @@ from opendm import log
from opendm.system import run
from opendm import io
def classify(point_cloud):
def classify(point_cloud, max_threads=8):
tmp_output = io.related_file_path(point_cloud, postfix=".classified")
if os.path.isfile(tmp_output):
os.remove(tmp_output)
@ -16,7 +16,7 @@ def classify(point_cloud):
name="model.bin")
if model is not None:
run('pcclassify "%s" "%s" "%s" -u -s 2,64' % (point_cloud, tmp_output, model))
run('pcclassify "%s" "%s" "%s" -u -s 2,64' % (point_cloud, tmp_output, model), env_vars={'OMP_NUM_THREADS': max_threads})
if os.path.isfile(tmp_output):
os.remove(point_cloud)

Wyświetl plik

@ -58,7 +58,7 @@ class ODMDEMStage(types.ODM_Stage):
)
log.ODM_INFO("Classifying {} using OpenPointClass (2/2)".format(dem_input))
classify(dem_input)
classify(dem_input, args.max_concurrency)
with open(pc_classify_marker, 'w') as f:
f.write('Classify: smrf\n')