OpenCV4.5.4训练级联分类器的代码,用pycharm写代码

时间: 2024-04-30 07:21:00 浏览: 16
以下是使用PyCharm编写OpenCV 4.5.4级联分类器训练的代码: 1. 安装OpenCV 4.5.4和numpy库 ``` pip install opencv-python==4.5.4.58 numpy ``` 2. 准备训练数据 将正样本和负样本图片分别放入两个文件夹中,并将文件夹路径记录下来。 3. 编写代码 ```python import cv2 import numpy as np # 设置正样本和负样本文件夹路径 pos_path = "path/to/positive/folder" neg_path = "path/to/negative/folder" # 设置输出文件夹路径和级联分类器名称 output_folder_path = "path/to/output/folder" cascade_name = "my_cascade.xml" # 设置级联分类器的宽度和高度 width, height = 24, 24 # 准备训练数据 pos_images = [] neg_images = [] # 加载正样本 for filename in os.listdir(pos_path): image = cv2.imread(os.path.join(pos_path, filename), cv2.IMREAD_GRAYSCALE) if image is not None: pos_images.append(image) # 加载负样本 for filename in os.listdir(neg_path): image = cv2.imread(os.path.join(neg_path, filename), cv2.IMREAD_GRAYSCALE) if image is not None: neg_images.append(image) # 设置正样本和负样本的标签 pos_labels = np.ones(len(pos_images), np.int32) neg_labels = np.zeros(len(neg_images), np.int32) # 将正样本和负样本合并 images = np.concatenate((pos_images, neg_images)) labels = np.concatenate((pos_labels, neg_labels)) # 创建Haar特征分类器 haar_cascade = cv2.CascadeClassifier("path/to/haarcascade_frontalface_default.xml") # 计算Haar特征 features = [] for image in images: feature = haar_cascade.detectMultiScale(image, 1.1, 5) if len(feature) > 0: feature = feature[0] feature = cv2.resize(image[feature[1]:feature[1]+feature[3], feature[0]:feature[0]+feature[2]], (width, height)) features.append(feature) # 将Haar特征转换为numpy数组 features = np.array(features) # 将Haar特征和标签分为训练集和测试集 train_features, test_features, train_labels, test_labels = train_test_split(features, labels, test_size=0.2) # 创建级联分类器 cascade = cv2.CascadeClassifier() # 训练级联分类器 cascade.train(train_features, train_labels, width=width, height=height) # 评估级联分类器 accuracy = cascade.evaluate(test_features, test_labels) # 保存级联分类器 cascade.save(os.path.join(output_folder_path, cascade_name)) # 打印准确度 print("Accuracy:", accuracy) ``` 以上代码中,我们首先设置了正样本和负样本的文件夹路径,并设置了输出文件夹路径和级联分类器名称。然后,我们加载了正样本和负样本的图片,并将它们分别存储在两个列表中。接下来,我们将正样本和负样本的标签设置为1和0,并将它们合并为一个标签列表。然后,我们使用Haar特征分类器计算Haar特征,并将其转换为numpy数组。最后,我们将Haar特征和标签分为训练集和测试集,并使用级联分类器训练数据。在训练完成后,我们评估了级联分类器的准确度,并将其保存到指定的文件夹中。

相关推荐

最新推荐

recommend-type

Opencv中imwrite函数源代码

Opencv中imwrite函数源代码,Opencv中imwrite函数源代码,Opencv中imwrite函数源代码
recommend-type

Python + OpenCV 实现LBP特征提取的示例代码

主要介绍了Python + OpenCV 实现LBP特征提取的示例代码,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
recommend-type

python+opencv实现车牌定位功能(实例代码)

主要介绍了python+opencv实现车牌定位功能,需要实现对给定的车牌进行车牌识别,本文通过实例代码讲解,需要的朋友可以参考下
recommend-type

OpenCV 表盘指针自动读数的示例代码

主要介绍了OpenCV 表盘指针自动读数的示例代码,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
recommend-type

Python 使用Opencv实现目标检测与识别的示例代码

主要介绍了Python 使用Opencv实现目标检测与识别的示例代码,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
recommend-type

zigbee-cluster-library-specification

最新的zigbee-cluster-library-specification说明文档。
recommend-type

管理建模和仿真的文件

管理Boualem Benatallah引用此版本:布阿利姆·贝纳塔拉。管理建模和仿真。约瑟夫-傅立叶大学-格勒诺布尔第一大学,1996年。法语。NNT:电话:00345357HAL ID:电话:00345357https://theses.hal.science/tel-003453572008年12月9日提交HAL是一个多学科的开放存取档案馆,用于存放和传播科学研究论文,无论它们是否被公开。论文可以来自法国或国外的教学和研究机构,也可以来自公共或私人研究中心。L’archive ouverte pluridisciplinaire
recommend-type

MATLAB柱状图在信号处理中的应用:可视化信号特征和频谱分析

![matlab画柱状图](https://img-blog.csdnimg.cn/3f32348f1c9c4481a6f5931993732f97.png) # 1. MATLAB柱状图概述** MATLAB柱状图是一种图形化工具,用于可视化数据中不同类别或组的分布情况。它通过绘制垂直条形来表示每个类别或组中的数据值。柱状图在信号处理中广泛用于可视化信号特征和进行频谱分析。 柱状图的优点在于其简单易懂,能够直观地展示数据分布。在信号处理中,柱状图可以帮助工程师识别信号中的模式、趋势和异常情况,从而为信号分析和处理提供有价值的见解。 # 2. 柱状图在信号处理中的应用 柱状图在信号处理
recommend-type

解释这行代码 c = ((double)rand() / RAND_MAX) * (a + b - fabs(a - b)) + fabs(a - b);

这行代码是用于生成 a 和 b 之间的随机数。首先,它使用 rand() 函数生成一个 [0,1) 之间的随机小数,然后将这个小数乘以 a、b 范围内的差值,再加上 a 和 b 中的较小值。这可以确保生成的随机数大于等于 a,小于等于 b,而且不会因为 a 和 b 之间的差距过大而导致难以生成足够多的随机数。最后,使用 fabs() 函数来确保计算结果是正数。
recommend-type

JSBSim Reference Manual

JSBSim参考手册,其中包含JSBSim简介,JSBSim配置文件xml的编写语法,编程手册以及一些应用实例等。其中有部分内容还没有写完,估计有生之年很难看到完整版了,但是内容还是很有参考价值的。