颜色分类leetcode color detector

wwwfxiangnet 1 0 zip 2024-10-06 10:10:31

颜色分类leetcode SVM颜色检测器这是一个OpenCV程序,可以检测图像中对象的颜色。这是我们对街道上汽车颜色进行分类的学校项目的一部分,但您也可以在自己的数据集上进行训练。

安装:您可以通过以下命令使用CMake构建此项目:

mkdir build

cd build

cmake ..

make

这会将代码编译到构建文件夹中。程序“train”和“test”可以简单地运行./test。

“测试”应用程序将加载一个预训练的SVM模型(“modell.xml”),该模型已经在识别汽车颜色方面进行了训练。它将加载测试图像(“data/test.jpg”)并对颜色进行分类。

不幸的是,由于版权原因,我无法发布训练数据集。

训练你自己的模型:如果你想训练自己的模型来检测物体的颜色,你需要做两个步骤:

  1. 调整“colorDetector.cpp”中的initClasses()方法。在这里你需要输入你想要训练的颜色的名称(例如“蓝色”、“黑色”等)。

用户评论
请输入评论内容
评分:
暂无评论