Parcourir la source

default resize

Jérôme BUISINE il y a 3 ans
Parent
commit
6caaae83ee
1 fichiers modifiés avec 1 ajouts et 0 suppressions
  1. 1 0
      train_lstm_weighted.py

+ 1 - 0
train_lstm_weighted.py

@@ -84,6 +84,7 @@ def build_input(df, seq_norm, p_chanels):
                     img = cv2.imread(img_path)
                 else:
                     img = cv2.imread(img_path, cv2.IMREAD_GRAYSCALE)
+                    img = cv2.resize(img, (50, 50))
                 
                 # normalization of images
                 seq_elems.append(np.array(img, 'float16') / 255.)