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

素颜马尾好姑娘i 2022-06-17 14:43 236阅读 0赞

使用时更改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输入反斜杠’\\’,需要用双反斜杠’\\\\’