[發明專利]一種多類別圖像轉換與對比的超高清片源色域檢測方法

有效

• 申請號: 201910971340.1
• 申請日: 2019-10-14
• 公開（公告）號: CN110781932B
• 公開（公告）日: 2022-03-11
• 發明（設計）人: 周蕓;胡瀟;郭曉強;李小雨;魏建超;梁琳琳
• 申請（專利權）人: 國家廣播電視總局廣播電視科學研究院;北京博雅睿視科技有限公司
• 主分類號: G06V10/764
• 分類號: G06V10/764;G06V10/82;G06K9/62;G06N3/04;G06N3/08
• 代理公司: 天津盛理知識產權代理有限公司 12209
• 代理人: 王利文

1.一種多類別圖像轉換與對比的超高清片源色域檢測方法，其特征在于包括以下步驟：

步驟1、構造多類別圖像轉換與對比的色域檢測網絡，用于對BT.709和BT.2020兩種視頻片源的色域進行檢測；

步驟2、構建圖像轉換與對比層，對輸入圖像進行轉換，增加不同類別特征間的參照和對比；

步驟3、使用殘差網絡作為初始網絡，將原始輸入圖像和轉換后的圖像同時輸入初始網絡；

步驟4、對色域檢測網絡進行訓練，得到BT.709和BT.2020兩類別的色域分類模型；

步驟5、將超高清片源圖像輸入到BT.709和BT.2020兩類別的色域分類模型中，由該色域分類模型輸出該圖像屬于兩類別的概率，最終根據概率大小判別圖像的色域類別；

所述步驟1構建的色域檢測網絡包括圖像轉換與對比層、五個殘差單元、全連接層和輸出層；原始輸入圖像經過圖像轉換與對比層進行圖像轉換操作，然后將圖像轉換與對比層輸出與原始輸入圖像一同送入五個殘差單元，每個殘差單元對圖像進行下采樣，輸出尺寸在長、寬上均減半，經過五個殘差單元后輸出特征圖再經過全連接層，得到一維特征向量作為輸入圖像的最終特征表示；輸出層用于輸出BT.709和BT.2020兩種色域分類；

所述步驟2的具體實現方法為：將BT.709和BT.2020的圖像均按照BT.709轉BT.2020的方法進行轉換，轉換后的BT.709圖像具備BT.2020的特征，轉換后的BT.2020圖像不具備BT.709的特征。

2.根據權利要求1所述的一種多類別圖像轉換與對比的超高清片源色域檢測方法，其特征在于：所述輸入圖像尺寸為224×224×3，五個殘差單元輸出的圖像尺寸為7×7，全連接層得到的一維特征向量為1×512。

3.根據權利要求1所述的一種多類別圖像轉換與對比的超高清片源色域檢測方法，其特征在于：所述殘差網絡采用ResNet34殘差網絡。

4.根據權利要求1所述的一種多類別圖像轉換與對比的超高清片源色域檢測方法，其特征在于：所述步驟4的具體實現方法為：輸入圖像首先被劃分為224×224×3像素尺寸，然后以批量大小為64輸入網絡進行訓練，迭代100epochs，最終得到訓練好的BT.709和BT.2020兩類別的色域分類模型。