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 fog_model.fog_model import darkchannel, sobel
#from fog_model import darkchannel, sobel
import pandas as pd
import joblib
import numpy as np
import os
import glob
#이미지 주소
paths = os.getcwd()
class fog_classfication():
def __init__(self):
#self.model=joblib.load(paths+'\\fog_model\\fog_model.pkl')
self.model=joblib.load(paths+'\\fog_model\\fog_model.pkl')
def predict(self,img):
svclassifier_from_pickle = self.model
alpha_map,result=darkchannel(img)
edge = sobel(img)
value=np.array([[result,edge]])
Y_pred = svclassifier_from_pickle.predict(value)
if Y_pred == 0:
return "fog"
else:
return "normal"