前提・実現したいこと
PythonでChainerを使用して数字認識を行いたいと思っています。CNNで学習済みのモデルをロードし、Eキーを押すと数字を判別してくれるプログラムなのですが、Eキーを押すと以下のエラーメッセージが出てフリーズしてしまいます。どこをどう直せばよいのか全く分からないので教えていただきたいです。よろしくお願い致します。
発生している問題・エラーメッセージ
TypeError: __call__() missing 1 required positional argument: 'x'
該当のソースコード
import cv2 import numpy as np from chainer import Chain, serializers import chainer.functions as F import chainer.links as L class CNN(Chain): def __init__(self, train=True): super(CNN, self).__init__( conv1=L.Convolution2D(1, 32, 5), conv2=L.Convolution2D(32, 64, 5), l1=L.Linear(1024, 10), ) self.train = train def __call__(self, x): h = F.max_pooling_2d(F.relu(self.conv1(x)), 2) h = F.max_pooling_2d(F.relu(self.conv2(h)), 2) return self.l1(h) def preprocessing(img): img = img[190:290,270:370] img = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY) img = cv2.GaussianBlur(img, (3, 3), 0) img = cv2.resize(img, (28, 28)) res, img = cv2.threshold(img, 70 , 255, cv2.THRESH_BINARY) img = 255 - img img = img.astype(np.float32) cv2.imwrite("img.jpg",img) img /= 255 img = np.array(img).reshape(1,784) return img def main(): # 学習済みモデルの読み込み net = L.Classifier(CNN()) serializers.load_npz('my_mnist.model', net) #Webカメラの映像表示 capture = cv2.VideoCapture(0) if capture.isOpened() is False: raise("IO Error") while True: #Webカメラの映像とりこみ ret, image = capture.read() if ret == False: continue #Webカメラの映像表示 cv2.rectangle(image,(270,190),(370,290),(0,0,255),3) cv2.imshow("Capture", image) k = cv2.waitKey(10) #Eキーで処理実行 if k == 101: img = preprocessing(image) num = net(img) #cv2.imwrite("img.jpg",img) print(num.data) print(np.argmax(num.data)) #ESCキーでキャプチャー画面を閉じる if k == 27: break cv2.destroyAllWindows() if __name__ == '__main__': main()
試したこと
ここに問題に対して試したことを記載してください。
補足情報(FW/ツールのバージョンなど)
ここにより詳細な情報を記載してください。
あなたの回答
tips
プレビュー