前提・実現したいこと
python初心者です。
OpenCVを用いて映像から特定の場面だけを抽出するプログラムを本を参考にして書いています。
その中でまず、画像にうつっている物体を学習するプログラムについてです。
特定の物体が移っている場面と映っていない場面の画像をそれぞれ100枚用意し、違うファイルに保存しています。
今回でいえば特定の物体が映っているものを"scrum_100"映っていないものを"no_scrum_100"としています。
発生している問題・エラーメッセージ
C:\Users\team_\Anaconda3\envs\tf1\lib\site-packages\sklearn\externals\joblib\__init__.py:15: FutureWarning: sklearn.externals.joblib is deprecated in 0.21 and will be removed in 0.23. Please import this functionality directly from joblib, which can be installed with: pip install joblib. If this warning is raised when loading pickled models, you may need to re-serialize those models with scikit-learn 0.21+. warnings.warn(msg, category=FutureWarning) Traceback (most recent call last): File "fish_train.py", line 28, in <module> read_dir(path_noscrum, 0) File "fish_train.py", line 22, in read_dir img = cv2.resize(img, image_size) cv2.error: OpenCV(4.2.0) C:\projects\opencv-python\opencv\modules\imgproc\src\resize.cpp:4045: error: (-215:Assertion failed) !ssize.empty() in function 'cv::resize'
該当のソースコード
python3
1import cv2 2import os, glob 3from sklearn.model_selection import train_test_split 4from sklearn import datasets, metrics 5from sklearn.ensemble import RandomForestClassifier 6from sklearn.metrics import accuracy_score 7from sklearn.externals import joblib 8 9# 画像の学習サイズやパスを指定 10image_size = (64, 32) 11path = os.path.dirname(os.path.abspath(__file__)) 12path_scrum = path + '/scrum_100' 13path_noscrum = path + '/no_scrum_100' 14x = [] # 画像データ 15y = [] # ラベルデータ 16 17# 画像データを読み込んで配列に追加 --- (*1) 18def read_dir(path, label): 19 files = glob.glob(path + "/*.jpg") 20 for f in files: 21 img = cv2.imread(f) 22 img = cv2.resize(img, image_size) 23 img_data = img.reshape(-1, ) # 一次元に展開 24 x.append(img_data) 25 y.append(label) 26 27# 画像データを読み込む 28read_dir(path_noscrum, 0) 29read_dir(path_scrum, 1) 30 31# データを学習用とテスト用に分割する --- (*2) 32x_train, x_test, y_train, y_test = train_test_split(x, y, test_size=0.2) 33 34# データを学習 --- (*3) 35clf = RandomForestClassifier() 36clf.fit(x_train, y_train) 37 38# 精度の確認 --- (*4) 39y_pred = clf.predict(x_test) 40print(accuracy_score(y_test, y_pred)) 41 42# データを保存 --- (*5) 43joblib.dump(clf, 'scrum.pkl') 44
試したこと
エラーについて調べましたが解決しませんでした。
補足情報(FW/ツールのバージョンなど)
本のサンプルコードをダウンロードし、パス名などを書き換えたものになります。
本の状況と違うこととしては、本ではファイルに保存されている画像のサイズがバラバラであるのに対し、
今回私が用意した画像は1920*720で一定です。
どうかよろしくお願いします。
バッドをするには、ログインかつ
こちらの条件を満たす必要があります。