LoginSignup
0
0

More than 5 years have passed since last update.

HOG特徴量とSVMの記事の差分

Last updated at Posted at 2017-12-17

HOG特徴量とSVMを使った自動車の検出

こちらの記事のコードを動かしてみましたが、何点か修正が必要だったので差分だけ記載します。

環境構築

# Python2.*
sudo python -m pip install scikit-image sklearn opencv-python

# Python3.*
pip install scikit-image sklearn opencv-python

全てのコード
print "..." は全て print("...")に書き換え

1つ目、3つ目のコード

<     fd = hog(neg_roi, orientations=9, pixels_per_cell=(6,6),cells_per_block=(3,3), visualise=False)
---
>     fd = hog(neg_roi, orientations=9, pixels_per_cell=(6,6),cells_per_block=(3,3), visualise=False, block_norm='L2')

hogの引数にblock_norm='L2'を追加

2つ目のコード
sklearn.grid_search はDeprecatedなので、sklearn.model_selectionを使用する。また不要なimportを削除

import numpy as np
from sklearn.model_selection import GridSearchCV
from sklearn import svm
from sklearn.externals import joblib

3つ目のコード
タブとスペースが混ざっているのでスペースに統一

test_img = resize(test_img, img_size, mode='constant')

引数にmode='constant'を追加

fd = hog(... の次の行に fd = fd.reshape(1, -1) が必要

hogの引数にblock_norm='L2'を追加

これでPython2でもPython3でも動くはず。

実行結果
Screen Shot 2017-12-17 at 18.12.00.png

0
0
0

Register as a new user and use Qiita more conveniently

  1. You get articles that match your needs
  2. You can efficiently read back useful information
  3. You can use dark theme
What you can do with signing up
0
0