Quellcode durchsuchen

input normalization

Jérôme BUISINE vor 4 Jahren
Ursprung
Commit
7b6a3ed93b
1 geänderte Dateien mit 1 neuen und 1 gelöschten Zeilen
  1. 1 1
      train_lstm_weighted.py

+ 1 - 1
train_lstm_weighted.py

@@ -50,7 +50,7 @@ def build_input(df, seq_norm):
             for img_path in column:
                 img = Image.open(img_path)
                 # seq_elems.append(np.array(img).flatten())
-                seq_elems.append(np.array(img))
+                seq_elems.append(np.array(img) / 255.)
 
             #seq_arr.append(np.array(seq_elems).flatten())
             seq_arr.append(np.array(seq_elems))