Skip to content

luanchenhui/HOGSVM

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

7 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

HogSvm

目标

实现对人头的检测并尝试用方框标出来。

使用方法

  1. 解压后,rawdata是要放置的数据。_0表示负样本,_1表示正样本。无需resize。datamake.py文件会进行resize操作
  2. 运行datamake.py文件后,文件会将rawdata中的数据图片放置到data文件夹下。(如data不存在会自动新建data文件夹)
  3. 运行data/datamk.py,会将data文件夹下的图片分类放置到/data/images/neg_position或/data/images/pos_position下。之后的特征提取与保存时从这两个文件下读文件。
  4. 运行object_detector/extract_features.py提取HOG并保存。
  5. 运行object_detector/train_svm.py进行支持向量机的训练并保存模型。
  6. 运行detect_detector/detect.py进行对test_image_mine文件夹的测试图片的检测,并展示效果。

其他说明:

  • inforce.py用于增强,简单修改代码中的字符串匹配可以改为是复制正样本还是复制负样本。
  • mname.py(manage name)用于管理rawdata的图片名称。虽然图片名称毫无影响。只要_1还是_0正确标注就行。
  • getsize.py 用于获取rawdata中图片的分辨率。最初是用于辨认测试图片中应设置的窗口大小。生成了img_shape.txt.
  • datamk.py与inforce.py因为调用的是linux中的cp命令进行的复制文件,所以需要在WSL下运行(windows subsystem for linux)可以在Microsoft Store中安装。或者在Linux下或其他terminal下运行。(斜眼笑)当然,最简单的还是自己改下代码,替换复制操作就好了。

About

A svm for head-detection using Hog

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Jupyter Notebook 94.3%
  • Python 5.7%