File name
Commit message
Commit date
File name
Commit message
Commit date
File name
Commit message
Commit date
import pickle
from sklearn.metrics import *
from model import darkchannel, sobel
import pandas as pd
import joblib
import numpy as np
#이미지 주소
def predict(X_test):
svclassifier_from_pickle = joblib.load('fog.pkl')
Y_pred = svclassifier_from_pickle.predict(X_test)
if Y_pred == 0:
print("fog")
else:
print("normal")
#print("정확도 : ", accuracy_score(Y_test,Y_pred))
#print(confusion_matrix(Y_test,Y_pred))
#print(classification_report(Y_test,Y_pred))
img = "C:/Users/lim/Desktop/data/85CF_HD_20211018_038932.jpg"
alpha_map,result=darkchannel(img)
edge = sobel(img)
value=np.array([[result,edge]])
predict(value)