From d306853c42fa06cde95c502f8915de0b04043bcd Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Thu, 20 Oct 2022 15:16:24 +0100 Subject: [PATCH] use right daataset --- aimodel/src/subcommands/train_predict.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/aimodel/src/subcommands/train_predict.py b/aimodel/src/subcommands/train_predict.py index f9fcba5..71f17b6 100644 --- a/aimodel/src/subcommands/train_predict.py +++ b/aimodel/src/subcommands/train_predict.py @@ -11,7 +11,7 @@ import tensorflow as tf from lib.vis.segmentation_plot import segmentation_plot from lib.io.handle_open import handle_open from lib.ai.RainfallWaterContraster import RainfallWaterContraster -from lib.dataset.dataset import dataset_predict +from lib.dataset.dataset_segmenter import dataset_predict from lib.io.find_paramsjson import find_paramsjson from lib.io.readfile import readfile from lib.vis.segmentation_plot import segmentation_plot