Skip to content

Commit b2c8703

Browse files
authored
[Fix] Fix random color for Palette (#2356)
1 parent 8b3346c commit b2c8703

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

demo/csrc/cpp/utils/palette.h

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -82,6 +82,7 @@ inline Palette Palette::get(int n) {
8282
std::vector<cv::Point3f> centers(n);
8383
cv::Mat c_mat(centers, false);
8484
cv::Mat s_mat(samples, false);
85+
c_mat = c_mat.reshape(1, {n, 3}); // CV_32FC3 -> CV_32FC1 for cv::kmeans output
8586
cv::kmeans(s_mat, n, indices, cv::TermCriteria(cv::TermCriteria::Type::COUNT, 10, 0), 1,
8687
cv::KMEANS_PP_CENTERS, c_mat);
8788
Palette p;

0 commit comments

Comments
 (0)