sbrl-github/research-rainfallradar
1
0
Fork 0
mirror of https://github.com/sbrl/research-rainfallradar synced 2024-11-25 18:33:01 +00:00
Code Issues Projects Releases Packages Wiki Activity

missing import

Browse source
This commit is contained in:
Starbeamrainbowlabs 2023-01-05 18:26:33 +00:00
parent 0d4cc63b76
commit 6a4f68a055
Signed by: sbrl
GPG key ID: 1BE5172E637709C2
1 changed files with 2 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
aimodel/src/deeplabv3_plus_test_rainfall.py
View file

@ -15,6 +15,8 @@ import matplotlib.pyplot as plt
import tensorflow as tf 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 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 BATCH_SIZE = int(os.environ["BATCH_SIZE"]) if "BATCH_SIZE" in os.environ else 64
NUM_CLASSES = 2 NUM_CLASSES = 2