草庐IT

c++ - OpenCV 中的快速颜色量化

coder 2023-11-14 原文

如何使用 OpenCV (+ C++) 以最快的方式减少图像中不同颜色的数量?我不想要完整的代码。我已经在使用 kmeans 了,但速度不是很快。这是我的代码中缓慢的部分:

kmeans(samples, clusterCount, labels,
    TermCriteria(TermCriteria::EPS + TermCriteria::COUNT, 10, 10.0),
    1, KMEANS_RANDOM_CENTERS, centers);

这段代码需要几秒钟的时间来处理,这对我来说非常慢。我为此使用了 Matlab( rgb2ind ),速度很快。几乎 0.01 秒。

我想将我的代码用于用户期望程序快速的生产环境。

有没有替代 kmeans 的颜色量化方法?有什么方法可以更快地运行 kmeans(我不这么认为,因为我尝试了许多不同的参数)?

编辑:
原来颜色量化是一个非常复杂的话题,需要时间来编写一个好的优化的。我决定使用 Magick++ (ImageMagick API)为了这。
因此,我还没有尝试过 Cris Luengo 的新(编辑)答案。但我将其标记为答案(也请查看评论),以免其他人认为此问题未得到解答。

最佳答案

有很多方法可以量化颜色。这里我描述四个。
均匀量化
这里我们使用颜色均匀分布的颜色图,无论它们是否存在于图像中。在 MATLAB 中,你会写

qimg = round(img*(N/255))*(255/N);
将每个 channel 量化为 N级别(假设输入在 [0,255] 范围内。您也可以使用 floor ,这在某些情况下更合适。这会导致 N^3 不同的颜色。例如,使用 N=8 您会得到 512 种独特的 RGB 颜色.
K均值聚类
这是生成自适应调色板的“经典”方法。显然,它将是最昂贵的。 OP 正在对所有像素的集合应用 k 均值。相反,k-means 可以应用于颜色直方图。过程是相同的,但不是 1000 万个数据点(现在的典型图像),您可能只有 32^3 = 33000 个。在处理自然照片时,由减少 bin 数量的直方图引起的量化在这里几乎没有影响。如果您要量化具有有限颜色集的图形,则不需要进行 k 均值聚类。
您对所有像素进行一次遍历以创建直方图。接下来,您运行常规 k 均值聚类,但使用直方图箱。每个数据点现在也有一个权重(该 bin 内的像素数),您需要考虑到这一点。算法中确定聚类中心的步骤会受到影响。您需要计算数据点的加权平均值,而不是常规平均值。
结果受初始化的影响。
八叉树量化
八叉树是一种用于空间索引的数据结构,其中通过将每个轴切成两半,将体积递归地划分为 8 个子体积。因此,树由节点组成,每个节点有 8 个子节点。对于颜色量化,RGB 立方体由八叉树表示,并计算每个节点的像素数(这相当于构建颜色直方图,并在其上构建八叉树)。接下来,叶节点被移除,直到留下所需数量的叶节点。删除叶节点一次发生 8 个,这样上一层的节点就变成了叶节点。选择修剪哪些节点有不同的策略,但它们通常围绕修剪像素数低的节点。
这是 Gimp 使用的方法。
因为八叉树总是从中间 split 节点,所以它不如 k-means 聚类或 next 方法灵活。
最小方差量化
MATLAB's rgb2ind ,OP 提到,做均匀量化和他们称之为“最小方差量化”的东西:

Minimum variance quantization cuts the RGB color cube into smaller boxes (not necessarily cubes) of different sizes, depending on how the colors are distributed in the image.


我不确定这意味着什么。 This page没有给出更多信息,但它有一个看起来像 RGB 立方体的 k-d 树分区的图形。 K-d 树是空间索引结构,它递归地将空间数据分成两半。在每一层,您选择分离度最大的维度,然后沿该维度拆分,生成一个额外的叶节点。与八叉树相反, split 可以发生在最佳位置,而不是在节点的中间。
使用空间索引结构(k-d 树或八叉树)的优点是颜色查找非常快。您从根开始,根据 R、G 或 B 值做出二元决策,直到到达叶节点。不需要像 k-means 那样计算到每个原型(prototype)集群的距离。
【两周后编辑】我一直在想一个可能的实现,而came up with one .这是算法:
  • 全彩色直方图被认为是一个分区。这将是 k-d 树的根,它现在也是叶节点,因为还没有其他节点。
  • 创建优先队列。它包含 k-d 树的所有叶节点。优先级由分区沿一个轴的方差减去两半的方差(如果我们要沿该轴拆分分区)给出。选择分割位置使得两半的方差最小(使用 Otsu 算法)。也就是说,优先级越大,我们通过拆分减少的总方差就越多。对于每个叶节点,我们为每个轴计算这个值,并使用最大的结果。
  • 我们处理队列上的分区,直到我们拥有所需的分区数:
  • 我们沿着轴和在确定优先级时计算的位置分割具有最高优先级的分区。
  • 我们计算两半的优先级,并将它们放在队列中。


  • 以这种方式描述时,这是一个相对简单的算法,the code有点复杂,因为我试图使其高效但通用。
    比较
    在 256x256x256 RGB 直方图上,我得到了这些时间比较 k-means 聚类和这个新算法:


    # 簇
    kmeans (s)
    最小变量


    5
    3.98
    0.34

    20
    17.9
    0.48

    50
    220.8
    0.59


    请注意,随着集群数量的增加,k-means 需要更多的迭代,因此指数时间增加。通常人们不会使用这么大的直方图,我希望拥有大量数据以使时序更加稳健。
    以下是应用于测试图像的这三种方法的示例:
    输入:

    制服搭配 N=4导致多达 64 种不同的颜色 [使用 N=2得到8种不同的颜色和其他方法相比,结果非常难看]:

    8 种颜色的 K 均值:

    具有 8 种颜色的新“最小方差”:

    我比 K 均值结果更喜欢最后一个结果,尽管它们非常相似。

    该程序说明了如何使用 DIPlib 进行颜色量化及其最小方差分区的实现:
    #include "diplib.h"
    #include "dipviewer.h"
    #include "diplib/simple_file_io.h"
    #include "diplib/histogram.h"
    #include "diplib/segmentation.h"
    #include "diplib/lookup_table.h"
    
    int main() {
       dip::Image input = dip::ImageRead( "/Users/cris/dip/images/flamingo.tif" );
       input.SetColorSpace( "RGB" ); // This image is linear RGB, not sRGB as assumed when reading RGB TIFFs.
    
       // Compute the color histogram.
       dip::Histogram hist( input, {}, { dip::Histogram::Configuration( 0.0, 255.0, 64 ) } );
    
       // Cluster the histogram, the output histogram has a label assigned to each bin.
       // Each label corresponds to one of the clusters.
       dip::uint nClusters = 8;
       dip::Image histImage = hist.GetImage(); // Copy with shared data
       dip::Image tmp;
       dip::CoordinateArray centers = dip::MinimumVariancePartitioning( histImage, tmp, nClusters );
       histImage.Copy( tmp ); // Copy 32-bit label image into 64-bit histogram image.
    
       // Find the cluster label for each pixel in the input image.
       dip::Image labels = hist.ReverseLookup( input );
    
       // The `centers` array contains histogram coordinates for each of the centers.
       // We need to convert these coordinates to RGB values by multiplying by 4 (=256/64).
       // `centers[ii]` corresponds to label `ii+1`.
       dip::Image lutImage( { nClusters + 1 }, 3, dip::DT_UINT8 );
       lutImage.At( 0 ) = 0; // label 0 doesn't exist
       for( dip::uint ii = 0; ii < nClusters; ++ii ) {
          lutImage.At( ii + 1 ) = { centers[ ii ][ 0 ] * 4, centers[ ii ][ 1 ] * 4, centers[ ii ][ 2 ] * 4 };
       }
    
       // Finally, we apply our look-up table mapping, painting each label in the image with
       // its corresponding RGB color.
       dip::LookupTable lut( lutImage );
       dip::Image output = lut.Apply( labels );
       output.SetColorSpace( "RGB" );
    
       // Display
       dip::viewer::ShowSimple( input, "input image" );
       dip::viewer::ShowSimple( output, "output image" );
       dip::viewer::Spin();
    }
    

    关于c++ - OpenCV 中的快速颜色量化,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/49710006/

    有关c++ - OpenCV 中的快速颜色量化的更多相关文章

    1. ruby - 如何从 ruby​​ 中的字符串运行任意对象方法? - 2

      总的来说,我对ruby​​还比较陌生,我正在为我正在创建的对象编写一些rspec测试用例。许多测试用例都非常基础,我只是想确保正确填充和返回值。我想知道是否有办法使用循环结构来执行此操作。不必为我要测试的每个方法都设置一个assertEquals。例如:describeitem,"TestingtheItem"doit"willhaveanullvaluetostart"doitem=Item.new#HereIcoulddotheitem.name.shouldbe_nil#thenIcoulddoitem.category.shouldbe_nilendend但我想要一些方法来使用

    2. ruby - 其他文件中的 Rake 任务 - 2

      我试图在一个项目中使用rake,如果我把所有东西都放到Rakefile中,它会很大并且很难读取/找到东西,所以我试着将每个命名空间放在lib/rake中它自己的文件中,我添加了这个到我的rake文件的顶部:Dir['#{File.dirname(__FILE__)}/lib/rake/*.rake'].map{|f|requiref}它加载文件没问题,但没有任务。我现在只有一个.rake文件作为测试,名为“servers.rake”,它看起来像这样:namespace:serverdotask:testdoputs"test"endend所以当我运行rakeserver:testid时

    3. ruby-on-rails - Ruby net/ldap 模块中的内存泄漏 - 2

      作为我的Rails应用程序的一部分,我编写了一个小导入程序,它从我们的LDAP系统中吸取数据并将其塞入一个用户表中。不幸的是,与LDAP相关的代码在遍历我们的32K用户时泄漏了大量内存,我一直无法弄清楚如何解决这个问题。这个问题似乎在某种程度上与LDAP库有关,因为当我删除对LDAP内容的调用时,内存使用情况会很好地稳定下来。此外,不断增加的对象是Net::BER::BerIdentifiedString和Net::BER::BerIdentifiedArray,它们都是LDAP库的一部分。当我运行导入时,内存使用量最终达到超过1GB的峰值。如果问题存在,我需要找到一些方法来更正我的代

    4. ruby-on-rails - Rails 3 中的多个路由文件 - 2

      Rails2.3可以选择随时使用RouteSet#add_configuration_file添加更多路由。是否可以在Rails3项目中做同样的事情? 最佳答案 在config/application.rb中:config.paths.config.routes在Rails3.2(也可能是Rails3.1)中,使用:config.paths["config/routes"] 关于ruby-on-rails-Rails3中的多个路由文件,我们在StackOverflow上找到一个类似的问题

    5. ruby-on-rails - Ruby on Rails : . 常量化 : wrong constant name error? - 2

      我正在使用这个:4.times{|i|assert_not_equal("content#{i+2}".constantize,object.first_content)}我之前声明过局部变量content1content2content3content4content5我得到的错误NameError:wrongconstantnamecontent2这个错误是什么意思?我很确定我想要content2=\ 最佳答案 你必须用一个大字母来调用ruby​​常量:Content2而不是content2。Aconstantnamestart

    6. ruby-on-rails - Rails - 一个 View 中的多个模型 - 2

      我需要从一个View访问多个模型。以前,我的links_controller仅用于提供以不同方式排序的链接资源。现在我想包括一个部分(我假设)显示按分数排序的顶级用户(@users=User.all.sort_by(&:score))我知道我可以将此代码插入每个链接操作并从View访问它,但这似乎不是“ruby方式”,我将需要在不久的将来访问更多模型。这可能会变得很脏,是否有针对这种情况的任何技术?注意事项:我认为我的应用程序正朝着单一格式和动态页面内容的方向发展,本质上是一个典型的网络应用程序。我知道before_filter但考虑到我希望应用程序进入的方向,这似乎很麻烦。最终从任何

    7. ruby-on-rails - Rails 3.2.1 中 ActionMailer 中的未定义方法 'default_content_type=' - 2

      我在我的项目中添加了一个系统来重置用户密码并通过电子邮件将密码发送给他,以防他忘记密码。昨天它运行良好(当我实现它时)。当我今天尝试启动服务器时,出现以下错误。=>BootingWEBrick=>Rails3.2.1applicationstartingindevelopmentonhttp://0.0.0.0:3000=>Callwith-dtodetach=>Ctrl-CtoshutdownserverExiting/Users/vinayshenoy/.rvm/gems/ruby-1.9.3-p0/gems/actionmailer-3.2.1/lib/action_mailer

    8. ruby-on-rails - 如何优雅地重启 thin + nginx? - 2

      我的瘦服务器配置了nginx,我的ROR应用程序正在它们上运行。在我发布代码更新时运行thinrestart会给我的应用程序带来一些停机时间。我试图弄清楚如何优雅地重启正在运行的Thin实例,但找不到好的解决方案。有没有人能做到这一点? 最佳答案 #Restartjustthethinserverdescribedbythatconfigsudothin-C/etc/thin/mysite.ymlrestartNginx将继续运行并代理请求。如果您将Nginx设置为使用多个上游服务器,例如server{listen80;server

    9. ruby-on-rails - Rails 应用程序中的 Rails : How are you using application_controller. rb 是新手吗? - 2

      刚入门rails,开始慢慢理解。有人可以解释或给我一些关于在application_controller中编码的好处或时间和原因的想法吗?有哪些用例。您如何为Rails应用程序使用应用程序Controller?我不想在那里放太多代码,因为据我了解,每个请求都会调用此Controller。这是真的? 最佳答案 ApplicationController实际上是您应用程序中的每个其他Controller都将从中继承的类(尽管这不是强制性的)。我同意不要用太多代码弄乱它并保持干净整洁的态度,尽管在某些情况下ApplicationContr

    10. ruby-on-rails - form_for 中不在模型中的自定义字段 - 2

      我想向我的Controller传递一个参数,它是一个简单的复选框,但我不知道如何在模型的form_for中引入它,这是我的观点:{:id=>'go_finance'}do|f|%>Transferirde:para:Entrada:"input",:placeholder=>"Quantofoiganho?"%>Saída:"output",:placeholder=>"Quantofoigasto?"%>Nota:我想做一个额外的复选框,但我该怎么做,模型中没有一个对象,而是一个要检查的对象,以便在Controller中创建一个ifelse,如果没有检查,请帮助我,非常感谢,谢谢

    随机推荐