[發(fā)明專利]一種圖像樣本處理方法、裝置及存儲(chǔ)介質(zhì)

有效

• 申請?zhí)枺?/td>: 202010403710.4
• 申請日: 2020-05-13
• 公開（公告）號(hào): CN111626342B
• 公開（公告）日: 2023-07-18
• 發(fā)明（設(shè)計(jì)）人: 賈玉虎
• 申請（專利權(quán)）人: OPPO廣東移動(dòng)通信有限公司
• 主分類號(hào): G06V10/774
• 分類號(hào): G06V10/774;G06V10/764;G06T3/00
• 代理公司: 北京派特恩知識(shí)產(chǎn)權(quán)代理有限公司 11270
• 代理人: 吳薇薇;張穎玲

本申請實(shí)施例公開了一種圖像樣本處理方法、裝置及存儲(chǔ)介質(zhì)，該包括：提取出原始數(shù)據(jù)集中第一圖像樣本的前景圖像，得到M張前景圖像；其中，M取正整數(shù)；采用預(yù)設(shè)的均衡化處理策略對所述M張前景圖像進(jìn)行均衡化處理，得到P張前景圖像；其中，P取大于M的整數(shù)；利用所述P張前景圖像和至少一張背景圖像構(gòu)建第二圖像樣本，并利用構(gòu)建的第二圖像樣本和所述第一圖像樣本組成目標(biāo)數(shù)據(jù)集。如此，通過對原始數(shù)據(jù)集中的前景圖像進(jìn)行均衡化處理，使不同類型或不同尺寸的前景圖像能夠均衡分布，再將均衡處理后的前景圖像與背景圖像合成，得到更多的圖像樣本，可以為檢測模型提高豐富且均衡的圖像樣本，從而提高檢測模型的訓(xùn)練效果。

技術(shù)領(lǐng)域

本申請涉及圖像處理技術(shù)，尤其涉及一種圖像樣本處理方法、裝置及存儲(chǔ)介質(zhì)。

背景技術(shù)

基于機(jī)器學(xué)習(xí)的圖像目標(biāo)檢測技術(shù)日漸成熟，已經(jīng)在諸多領(lǐng)域得到應(yīng)用，并發(fā)揮強(qiáng)大的技術(shù)效能。檢測模型在訓(xùn)練時(shí)通常利用一個(gè)公開的數(shù)據(jù)集或者自己制作的數(shù)據(jù)集來作為訓(xùn)練樣本，但現(xiàn)有數(shù)據(jù)集中不可避免的會(huì)存在圖像樣本類別不均衡的問題，數(shù)據(jù)集中圖像樣本類別分布不均衡會(huì)對檢測模型的收斂產(chǎn)生很大的影響，從而影響檢測模型的檢測效果。

發(fā)明內(nèi)容

為解決上述技術(shù)問題，本申請實(shí)施例期望提供一種圖像樣本處理方法、裝置及存儲(chǔ)介質(zhì)。

本申請的技術(shù)方案是這樣實(shí)現(xiàn)的：

第一方面，提供了一種種圖像樣本處理方法，所述方法包括：

提取出原始數(shù)據(jù)集中第一圖像樣本的前景圖像，得到M張前景圖像；其中，M取正整數(shù)；

采用預(yù)設(shè)的均衡化處理策略對所述M張前景圖像進(jìn)行均衡化處理，得到P張前景圖像；其中，P取大于M的整數(shù)；

利用所述P張前景圖像和至少一張背景圖像構(gòu)建第二圖像樣本，并利用構(gòu)建的第二圖像樣本和所述第一圖像樣本組成目標(biāo)數(shù)據(jù)集。

第二方面，提供了一種圖像樣本處理裝置，所述裝置包括：

提取單元，用于提取出原始數(shù)據(jù)集中第一圖像樣本的前景圖像，得到M張前景圖像；其中，M取正整數(shù)；

均衡單元，用于采用預(yù)設(shè)的均衡化處理策略對所述M張前景圖像進(jìn)行均衡化處理，得到P張前景圖像；其中，P取大于M的整數(shù)；

構(gòu)建單元，用于利用所述P張前景圖像和至少一張背景圖像構(gòu)建第二圖像樣本，并利用構(gòu)建的第二圖像樣本和所述第一圖像樣本組成目標(biāo)數(shù)據(jù)集。

第三方面，提供了一種計(jì)算機(jī)存儲(chǔ)介質(zhì)，其上存儲(chǔ)有計(jì)算機(jī)程序，其中，該計(jì)算機(jī)程序被處理器執(zhí)行時(shí)實(shí)現(xiàn)前述方法的步驟。

采用上述技術(shù)方案，提取出原始數(shù)據(jù)集中第一圖像樣本的前景圖像，得到M張前景圖像；其中，M取正整數(shù)；采用預(yù)設(shè)的均衡化處理策略對所述M張前景圖像進(jìn)行均衡化處理，得到P張前景圖像；其中，P取大于M的整數(shù)；利用所述P張前景圖像和至少一張背景圖像構(gòu)建第二圖像樣本，并利用構(gòu)建的第二圖像樣本和所述第一圖像樣本組成目標(biāo)數(shù)據(jù)集。如此，通過對原始數(shù)據(jù)集中的前景圖像進(jìn)行均衡化處理，使不同類型或不同尺寸的前景圖像能夠均衡分布，再將均衡處理后的前景圖像與背景圖像合成，得到更多的圖像樣本，可以為檢測模型提高豐富且均衡的圖像樣本，從而提高檢測模型的訓(xùn)練效果。

附圖說明

圖1為本申請實(shí)施例中圖像樣本處理方法的流程示意圖；

圖2為本申請實(shí)施例中原始數(shù)據(jù)集中不同尺度前景圖像的分布示意圖；

圖3為本申請實(shí)施例中原始數(shù)據(jù)集中不同類別樣本數(shù)量的分布示意圖；

圖4為本申請實(shí)施例中不同尺度前景圖像擴(kuò)充數(shù)量的分布示意圖；

圖5為本申請實(shí)施例中擴(kuò)充后的不同尺度前景圖像的第一分布示意圖；