From 6a4f68a0557a6565b63b99b7c288639f3d2d0443 Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Thu, 5 Jan 2023 18:26:33 +0000 Subject: [PATCH] missing import --- aimodel/src/deeplabv3_plus_test_rainfall.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/aimodel/src/deeplabv3_plus_test_rainfall.py b/aimodel/src/deeplabv3_plus_test_rainfall.py index d62cab8..839fe3a 100755 --- a/aimodel/src/deeplabv3_plus_test_rainfall.py +++ b/aimodel/src/deeplabv3_plus_test_rainfall.py @@ -15,6 +15,8 @@ import matplotlib.pyplot as plt import tensorflow as tf +from lib.dataset.dataset_mono import dataset_mono + IMAGE_SIZE = int(os.environ["IMAGE_SIZE"]) if "IMAGE_SIZE" in os.environ else 128 # was 512; 128 is the highest power of 2 that fits the data BATCH_SIZE = int(os.environ["BATCH_SIZE"]) if "BATCH_SIZE" in os.environ else 64 NUM_CLASSES = 2