diff --git a/modules/deeplearning/cnn_cld_frac.py b/modules/deeplearning/cnn_cld_frac.py index 95796539c82330372e49e6edd899dada6c305e68..312f42e3893646ef3c6197d520ac885bb7e1781f 100644 --- a/modules/deeplearning/cnn_cld_frac.py +++ b/modules/deeplearning/cnn_cld_frac.py @@ -235,7 +235,7 @@ class CNN: self.test_data_nda = None self.test_label_nda = None - self.n_chans = len(data_params) + 2 + self.n_chans = len(data_params) + 1 self.X_img = tf.keras.Input(shape=(None, None, self.n_chans)) # self.X_img = tf.keras.Input(shape=(36, 36, self.n_chans)) @@ -282,7 +282,7 @@ class CNN: # -------- tmp = input_data[:, label_idx, y_128, x_128] tmp = np.where(np.isnan(tmp), 0, tmp) # shouldn't need this - data_norm.append(tmp) + # data_norm.append(tmp) # --------- data = np.stack(data_norm, axis=3) data = data.astype(np.float32)