traincascade报错:“Train dataset for temp stage can not be filled. Branch training terminated ..."

最新推荐文章于 2021-03-30 11:38:47 发布

一把木剑

最新推荐文章于 2021-03-30 11:38:47 发布

阅读量1.9k

点赞数 7

分类专栏：视觉相关文章标签： OPENCV traincascade 分类器训练及测试

本文链接：https://blog.csdn.net/banzhuan133/article/details/82227772

版权

在使用OpenCV的traincascade进行分类器训练时，遇到'Train dataset for temp stage can not be filled. Branch training terminated ...'的错误。问题在于正样本描述文件需包含每个样本的位置和大小信息，而负样本则不需要。文章提供了解决方案，包括如何创建正负样本描述文件，并分享了一个用于测试分类器的简单脚本。

摘要由CSDN通过智能技术生成

正样本描述文件包括正样本文件名（包括绝对路径或相对路径），正样本数目以及各正样本在图片中的位置和大小，类似于

/home/nvidia/Desktop/PythonPic/pos/32.jpg 1 0 0 40 40
/home/nvidia/Desktop/PythonPic/pos/14.jpg 1 0 0 40 40
/home/nvidia/Desktop/PythonPic/pos/59.jpg 1 0 0 40 40
/home/nvidia/Desktop/PythonPic/pos/40.jpg 1 0 0 40 40
/home/nvidia/Desktop/PythonPic/pos/27.jpg 1 0 0 40 40

但是负样本根本没有样本在图片中的位置！因此不需要后面那一串描述！如下即可

/home/nvidia/Desktop/PythonPic/neg/68.jpg
/home/nvidia/Desktop/PythonPic/neg/100.jpg
/home/nvidia/Desktop/PythonPic/neg/65.jpg
/home/nvidia/Desktop/PythonPic/neg/114.jpg

写了两个简单的脚本来完成描述文件，负样本无需生成.vec文件

#pos.py
#!/usr/bin/env python2.7
# -*- coding: utf-8 -*-   
      
import os  
import sys
def file_name(file_dir):   
	L=[]   
	for root, dirs, files in os.walk(file_dir):  
	    for file in files: