您的位置:首页 > 编程语言 > Python开发

python生成opencv正样本和负样本描述文件

2017-04-28 17:00 771 查看
使用时更改rootdir即可,即修改文件路径即可,文件路径为你的正样本保存的目录或者负样本保存的目录,以下为正样本
pos.dat
生成的程序:

注意:为了保证不出错,在生成正样本和负样本后最好用编辑器打开看一下,删除最后一行的空行

#!/usr/bin env python
import os
rootdir = 'pos\\'
files = os.listdir(rootdir)
name = os.path.split(files[0])
with open(rootdir+'pos.dat','w+') as f:
for file in files:
name = os.path.split(file)
if name[1]=='pos.dat':
continue
print file
f.write("pos\\"+name[1]+' '+'1 0 0 30 30\n')


以下为生成负样本
neg.dat
的程序:

#!/usr/bin env python
import os
rootdir = 'neg\\'
files = os.listdir(rootdir)
name = os.path.split(files[0])
with open(rootdir+'neg.dat','w+') as f:
for file in files:
name = os.path.split(file)
if name[1]=='neg.dat':
continue
print file
f.write("neg\\"+name[1]+'\n')


生成正样本
pos.dat
neg.dat
后,将包含正样本的文件夹和负样本的文件夹以及
pos.dat
neg.dat
文件移动到
opencv\build\x64\vc12\bin\
目录下,好,接下来开始训练,请参考我的另一篇博客
http://blog.csdn.net/m0_38128647/article/details/71047944


附:windows下自动获取文件夹下文件名的命令
dir/s/b>pos.txt
,切换到所要操作的文件夹下,输入上边那个命令,即可将此文件夹下所有文件名写入pos.txt 文件里面

还有windows下python输入反斜杠’\’,需要用双反斜杠’\\’
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  python opencv