颜色分类leetcode color detector
颜色分类leetcode SVM颜色检测器这是一个OpenCV程序,可以检测图像中对象的颜色。这是我们对街道上汽车颜色进行分类的学校项目的一部分,但您也可以在自己的数据集上进行训练。
安装:您可以通过以下命令使用CMake构建此项目:
mkdir build
cd build
cmake ..
make
这会将代码编译到构建文件夹中。程序“train”和“test”可以简单地运行./test。
“测试”应用程序将加载一个预训练的SVM模型(“modell.xml”),该模型已经在识别汽车颜色方面进行了训练。它将加载测试图像(“data/test.jpg”)并对颜色进行分类。
不幸的是,由于版权原因,我无法发布训练数据集。
训练你自己的模型:如果你想训练自己的模型来检测物体的颜色,你需要做两个步骤:
-
调整“colorDetector.cpp”中的initClasses()方法。在这里你需要输入你想要训练的颜色的名称(例如“蓝色”、“黑色”等)。