Ulepszenie datasetu.

algorytm_biedronki.py

@@ -13,7 +13,7 @@ import os
 import cv2
 import matplotlib.pyplot as plt
 # %%
-os.chdir(r"C:\Users\pittb\Documents\biedronkapp\BiedronkApp\biedronki_dataset")
+os.chdir(r"C:\Users\pittb\Documents\biedronkapp_2.0\BiedronkApp\biedronki_dataset")
 train_path="train"
 test_path="test"
 # %%
@@ -133,3 +133,20 @@ accuracy_score(y_pred,test_y)
 
 print(classification_report(y_pred,test_y))
 # %%
+img_path = r"C:\Users\pittb\Documents\biedronkapp_2.0\BiedronkApp\biedronki_dataset\test\azjatyckie\78687-Biedronki-azjatyckie-Czy-sa-grozne-fot.-cocoparisienne---Pixabay.jpg"
+
+img_nump = np.array(cv2.imread(img_path))
+img_nump = cv2.resize(img_nump,(224,224))
+img_nump = img_nump/255.0
+img_nump = img_nump[np.newaxis,...]
+
+pr = model.predict(img_nump)
+print(pr)
+print(np.argmax(pr,axis=1))
+# %%
+# %%
+model.save('saved_model/my_model')
+
+# %%
+
+# %%