diff --git a/vhr_toolkit/view/vhrToolkit.py b/vhr_toolkit/view/vhrToolkit.py index c3f62dd..049505e 100755 --- a/vhr_toolkit/view/vhrToolkit.py +++ b/vhr_toolkit/view/vhrToolkit.py @@ -74,8 +74,8 @@ def main(): cMaskDir.mkdir(exist_ok=True) cMaskActualOutDir = cMaskDir / '5-toas' toaRegex = [str(toaDir / '*-toa.tif')] - - cmpl = CloudMaskPipeline(output_dir=cMaskDir, + + cmpl = CloudMaskPipeline(output_dir=cMaskDir, inference_regex_list=toaRegex) cmpl.predict()