diff options
| author | dkazanc <dkazanc@hotmail.com> | 2019-04-11 15:58:36 +0100 |
|---|---|---|
| committer | dkazanc <dkazanc@hotmail.com> | 2019-04-11 15:58:36 +0100 |
| commit | f736660128290df50aa9a431dfa40e96da600f0d (patch) | |
| tree | cf2808b727bac9d338c81fa62bade7457a819c52 /src/Python/ccpi | |
| parent | 2d436f37be6029d57d7f876d4c7c378ee712a11e (diff) | |
| download | regularization-f736660128290df50aa9a431dfa40e96da600f0d.tar.gz regularization-f736660128290df50aa9a431dfa40e96da600f0d.tar.bz2 regularization-f736660128290df50aa9a431dfa40e96da600f0d.tar.xz regularization-f736660128290df50aa9a431dfa40e96da600f0d.zip | |
further progress
Diffstat (limited to 'src/Python/ccpi')
| -rw-r--r-- | src/Python/ccpi/filters/regularisers.py | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/src/Python/ccpi/filters/regularisers.py b/src/Python/ccpi/filters/regularisers.py index 00afcc2..610907d 100644 --- a/src/Python/ccpi/filters/regularisers.py +++ b/src/Python/ccpi/filters/regularisers.py @@ -127,11 +127,13 @@ def NDF(inputData, regularisation_parameter, edge_parameter, iterations, raise ValueError ('GPU is not available') raise ValueError('Unknown device {0}. Expecting gpu or cpu'\ .format(device)) -def NDF_MASK(inputData, maskdata, diffuswindow, regularisation_parameter, edge_parameter, iterations, +def NDF_MASK(inputData, maskdata, select_classes, total_classesNum, diffuswindow, regularisation_parameter, edge_parameter, iterations, time_marching_parameter, penalty_type, tolerance_param, device='cpu'): if device == 'cpu': return NDF_MASK_CPU(inputData, maskdata, + select_classes, + total_classesNum, diffuswindow, regularisation_parameter, edge_parameter, @@ -142,6 +144,8 @@ def NDF_MASK(inputData, maskdata, diffuswindow, regularisation_parameter, edge_p elif device == 'gpu' and gpu_enabled: return NDF_MASK_CPU(inputData, maskdata, + select_classes, + total_classesNum, diffuswindow, regularisation_parameter, edge_parameter, |