以下のホームページを読んでいたところ、下記のようなコードが書かれていました。
どうして白黒反転を行う必要があるのですか。試しに#1までのコードのみの関数predict_digit()を使って写真を出したところ、単純に真っ黒な写真が表示されてしまいました。
https://note.com/dngri/n/n63d597962b9e
Python
1import cv2 2from sklearn.externals import joblib 3 4def predict_digit(filename): 5 clf = joblib.load("digits.pkl") 6 my_img = cv2.imread(filename) 7 my_img = cv2.cvtColor(my_img, cv2.COLOR_BGR2GRAY) 8 my_img = cv2.resize(my_img,(8,8)) 9 #白黒反転する? 10 my_img = 15 - my_img // 16 11 #1 12 my_img = my_img.reshape((-1,64)) 13 res = clf.predict(my_img) 14 return res[0] 15 16n = predict_digit("test.jpg") 17print("test = " + str(n))
Python
1from sklearn.model_selection import train_test_split 2from sklearn import datasets,svm,metrics 3from sklearn.metrics import accuracy_score 4digits = datasets.load_digits() 5x = digits.images 6y = digits.target 7#二次元配列を一次元配列に変換 8x = x.reshape((-1,64)) 9# データを学習用とテスト用に分割する 10x_train,x_test,y_train,y_test = train_test_split(x,y,test_size=0.2) 11# データを学習 12clf = svm.LinearSVC() 13clf.fit(x_train, y_train) 14 15from sklearn.externals import joblib 16joblib.dump(clf, 'digits.pkl')
回答1件
あなたの回答
tips
プレビュー