Преглед на файлове

Reduce data amount using float32 value

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