changed target image size

X-ray-Dawgz · Mar 18, 2020 · 4b9962b · 4b9962b
1 parent 68677b3
commit 4b9962b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/XRayDawgz/predict.py b/XRayDawgz/predict.py
@@ -11,7 +11,7 @@ def crystal_structure(path_to_image):
     cnn_model = keras.models.load_model('crystal_structure_classifier.h5')
     for file in os.listdir(path_to_image):
         predict_img = image.load_img(path_to_image + file,
-                target_size = (432,288))
+                target_size = (379,288))
         predict_img = image.img_to_array(predict_img)
         predict_img = np.expand_dims(predict_img, axis = 0)
         result = cnn_model.predict(predict_img)