Browse Source

Reduce data amount using float32 value

Jérôme BUISINE 3 years ago
parent
commit
e3e10a84c8
1 changed files with 1 additions and 1 deletions
  1. 1 1
      train_lstm_weighted.py

+ 1 - 1
train_lstm_weighted.py

@@ -85,7 +85,7 @@ def build_input(df, seq_norm, p_chanels):
                 else:
                     img = cv2.imread(img_path, cv2.IMREAD_GRAYSCALE)
                 
-                seq_elems.append(np.array(img) / 255.)
+                seq_elems.append(np.array(img, 'float32') / 255.)
 
             #seq_arr.append(np.array(seq_elems).flatten())
             seq_arr.append(np.array(seq_elems))