trial push

pothole-detection @ cec0185f

+Subproject commit cec0185ff75cabc37fe007c98fc8b89f6f74e727

scripts/Prediction.py

@@ -5,10 +5,8 @@ for i in tqdm(files):
     plt.figure()
     plt.imshow(X[:,:,::-1]) 
     plt.show()  
-
     X = np.array(X)
     X = np.expand_dims(X, axis=0)
-
     y_pred = np.round(model.predict(X))
     if y_pred[0][0] == 1:
         print("plain Road")

scripts/train.py

@@ -4,11 +4,7 @@ from sklearn.model_selection import KFold
 from sklearn.metrics import accuracy_score,precision_score,recall_score,confusion_matrix,roc_curve,roc_auc_score
 from sklearn.model_selection import GridSearchCV
 from sklearn.preprocessing import LabelEncoder
-
-
 # In[ ]:
-
-
 from keras.layers import Dropout, Flatten,Activation
 from keras.layers import Conv2D, MaxPooling2D, BatchNormalization
 from keras.layers import InputLayer