在区分多目标或者要对多个区域赋予不同颜色的时候,就需要准备多种颜色,常用的方法就是预先定义一个颜色数组,这里将通过随机数来创建这样的数组。
vector<Vec3b> colorTab;
for(int i = 0; i < train_sample_count; i++ )
{
int b = theRNG().uniform(0, 255);
int g = theRNG().uniform(0, 255);
int r = theRNG().uniform(0, 255);
colorTab.push_back(Vec3b((uchar)b, (uchar)g, (uchar)r));
}
上面的代码创建的颜色数组是用来给CV_8UC3类型的Mat赋值的,可以很方便的给多个区域块赋予不同的颜色。根据Mat的类型对上面的代码进行适当的修改就可以对各类型的Mat进行赋值了。当要对多目标进行区分时,如在多目标跟踪时将跟踪到的目标用不同颜色的矩形标出时,只需要将Vec3b换成与Mat类型相吻合的Scalar即可。
下面将以KNN来举一个简单的例子,简单起见取K=1。
int main()
{
int train_sample_count = 5;
int r, c;
RNG rng_state = RNG();
Mat trainData(train_sample_count, 2, CV_32FC1);
Mat trainClasses(train_sample_count, 1, CV_32FC1);
rng_state.fill(trainData, RNG::UNIFORM, Scalar(0, 0), Scalar(500, 500));
Mat img = Mat::zeros(500, 500, CV_8UC3);
for (int m = 0; m < train_sample_count; m++)
{
((float*)trainClasses.data)[m] = m;
}
CvKNearest knn(trainData, trainClasses, Mat(), false, 1);
Mat smmple(1, 2, CV_32FC1);
vector<Vec3b> colorTab;
for(int i = 0; i < train_sample_count; i++ )
{
int b = theRNG().uniform(0, 255);
int g = theRNG().uniform(0, 255);
int r = theRNG().uniform(0, 255);
colorTab.push_back(Vec3b((uchar)b, (uchar)g, (uchar)r));
}
for (r = 0; r < img.rows; r++)
{
for (c = 0; c < img.cols; c++)
{
((float*)smmple.data)[0] = c;
((float*)smmple.data)[1] = r;
float response = knn.find_nearest(smmple, 1);
img.at<Vec3b>(r, c) = colorTab[response];
}
}
for (r = 0; r < trainData.rows; r++)
{
float* ptr = (float*)trainData.ptr(r);
int x = cvRound(ptr[0]);
int y = cvRound(ptr[1]);
circle(img, Point(x, y), 2, Scalar(0, 255, 255), CV_FILLED);
}
imshow("img", img);
waitKey();
return 0;
}
效果图: