We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 8b3346c commit b2c8703Copy full SHA for b2c8703
demo/csrc/cpp/utils/palette.h
@@ -82,6 +82,7 @@ inline Palette Palette::get(int n) {
82
std::vector<cv::Point3f> centers(n);
83
cv::Mat c_mat(centers, false);
84
cv::Mat s_mat(samples, false);
85
+ c_mat = c_mat.reshape(1, {n, 3}); // CV_32FC3 -> CV_32FC1 for cv::kmeans output
86
cv::kmeans(s_mat, n, indices, cv::TermCriteria(cv::TermCriteria::Type::COUNT, 10, 0), 1,
87
cv::KMEANS_PP_CENTERS, c_mat);
88
Palette p;
0 commit comments