Jérôme BUISINE преди 4 години
родител
ревизия
7b6a3ed93b
променени са 1 файла, в които са добавени 1 реда и са изтрити 1 реда
  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))