浏览代码

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))