jha11aditya
3 years ago
8 changed files with 72 additions and 9 deletions
@ -0,0 +1,57 @@ |
|||
import os |
|||
import sys |
|||
import pandas as pd |
|||
import numpy as np |
|||
from skimage.io import imread, imshow, imsave |
|||
from skimage.transform import resize |
|||
from sklearn.preprocessing import StandardScaler |
|||
from sklearn.decomposition import PCA |
|||
from sklearn.model_selection import train_test_split |
|||
from sklearn import svm |
|||
from sklearn import metrics |
|||
import pickle |
|||
from skimage import img_as_ubyte |
|||
|
|||
tempdir = "predictor_temp" |
|||
|
|||
if len(sys.argv) < 3: |
|||
print("no input image") |
|||
exit(0) |
|||
|
|||
if len(sys.argv) < 2: |
|||
print("no input model") |
|||
exit(0) |
|||
|
|||
svm_model = pickle.load(open(sys.argv[1], 'rb')) |
|||
|
|||
os.system('mkdir -p ' + tempdir) |
|||
os.system("python3 face_cutter.py " + sys.argv[2] + " " + tempdir ) |
|||
img_files = [name for name in os.listdir(tempdir) if not os.path.isdir(os.path.join(tempdir, name)) ] |
|||
print(img_files) |
|||
cutf = tempdir +"/"+img_files[0] |
|||
os.system("python3 hogger.py " + cutf + " " + tempdir ) |
|||
os.system("rm " + cutf) |
|||
img_files = [name for name in os.listdir(tempdir) if not os.path.isdir(os.path.join(tempdir, name)) ] |
|||
hogf = tempdir + "/" + img_files[0] |
|||
|
|||
imgdat = imread(hogf, as_gray=True) |
|||
os.system("rm -rf " + tempdir) |
|||
imgdat = resize(imgdat, (64,64)) |
|||
|
|||
imgdat = img_as_ubyte(imgdat) |
|||
|
|||
|
|||
flat_imgdat = np.array( imgdat ).flatten() |
|||
print(flat_imgdat.shape) |
|||
X = np.array(flat_imgdat) |
|||
X = X.reshape(-1,1) |
|||
X = StandardScaler().fit_transform(X) |
|||
print(X.shape,X) |
|||
# pca = PCA(n_components=128) |
|||
# pcaofX = pca.fit_transform(X) |
|||
|
|||
|
|||
res = svm_model.predict(X.T) |
|||
|
|||
print("Prediction:",res) |
|||
|
After Width: | Height: | Size: 23 KiB |
After Width: | Height: | Size: 17 KiB |
After Width: | Height: | Size: 24 KiB |
After Width: | Height: | Size: 22 KiB |
After Width: | Height: | Size: 28 KiB |
Binary file not shown.
Loading…
Reference in new issue