如何通過(guò)深度學(xué)習(xí)，完成計(jì)算機(jī)視覺(jué)中的所有工作？

作者 | George Seif
譯者 | 天道酬勤，責(zé)編 | Carol
出品 | AI科技大本營(yíng)（ID:rgznai100）

你想做計(jì)算機(jī)視覺(jué)嗎？

如今，深度學(xué)習(xí)是必經(jīng)之路。大規(guī)模數(shù)據(jù)集以及深層卷積神經(jīng)網(wǎng)絡(luò)（CNN）的表征能力可提供超準(zhǔn)確和強(qiáng)大的模型。但目前仍然只有一個(gè)挑戰(zhàn)：如何設(shè)計(jì)模型？

像計(jì)算機(jī)視覺(jué)這樣廣泛而復(fù)雜的領(lǐng)域，解決方案并不總是清晰明了的。計(jì)算機(jī)視覺(jué)中的許多標(biāo)準(zhǔn)任務(wù)都需要特別考慮：分類、檢測(cè)、分割、姿態(tài)估計(jì)、增強(qiáng)和恢復(fù)以及動(dòng)作識(shí)別。盡管最先進(jìn)的網(wǎng)絡(luò)呈現(xiàn)出共同的模式，但它們都需要自己獨(dú)特的設(shè)計(jì)。

那么，我們?nèi)绾螢樗羞@些不同的任務(wù)建立模型呢？

作者在這里向你展示如何通過(guò)深度學(xué)習(xí)完成計(jì)算機(jī)視覺(jué)中的所有工作！

1、分類

計(jì)算機(jī)視覺(jué)中最出名的就是分類。圖像分類網(wǎng)絡(luò)從一個(gè)固定大小的輸入開(kāi)始。輸入圖像可以有任意數(shù)量的通道，但對(duì)于RGB圖像通常為3。在設(shè)計(jì)網(wǎng)絡(luò)時(shí)，分辨率在技術(shù)上可以是任意大小，只要足夠大到能夠支持在整個(gè)網(wǎng)絡(luò)中將要進(jìn)行的向下采樣量即可。例如，如果你對(duì)網(wǎng)絡(luò)內(nèi)的4個(gè)像素進(jìn)行向下采樣，則你的輸入大小至少應(yīng)為42= 16 x 16像素。

隨著深入網(wǎng)絡(luò)，當(dāng)我們嘗試壓縮所有信息并降至一維矢量表示形式時(shí)，空間分辨率將降低。為了確保網(wǎng)絡(luò)始終有能力將其提取的所有信息進(jìn)行處理，我們根據(jù)深度的比例增加特征圖的數(shù)量，來(lái)適應(yīng)空間分辨率的降低。也就是說(shuō)，我們?cè)谙蛳虏蓸舆^(guò)程中損失了空間信息，為了適應(yīng)這種損失，我們擴(kuò)展了特征圖來(lái)增加我們的語(yǔ)義信息。

在選擇了一定數(shù)量的向下采樣后，特征圖被矢量化并輸入到一系列完全連接的圖層中。最后一層的輸出與數(shù)據(jù)集中的類一樣多。

2、目標(biāo)檢測(cè)

目標(biāo)檢測(cè)器分為兩種：一級(jí)和二級(jí)。他們兩個(gè)都以錨框開(kāi)始。這些是默認(rèn)的邊界框。我們的檢測(cè)器將預(yù)測(cè)這些框與地面真相之間的差異，而不是直接預(yù)測(cè)這些框。
在二級(jí)檢測(cè)器中，我們自然有兩個(gè)網(wǎng)絡(luò)：框提議網(wǎng)絡(luò)和分類網(wǎng)絡(luò)?？蛱嶙h網(wǎng)絡(luò)在認(rèn)為很有可能存在物體的情況下為邊界框提供坐標(biāo)。再次，這些是相對(duì)于錨框。然后，分類網(wǎng)絡(luò)獲取每個(gè)邊界框中的潛在對(duì)象進(jìn)行分類。

在一級(jí)檢測(cè)器中，提議和分類器網(wǎng)絡(luò)融合為一個(gè)單一階段。網(wǎng)絡(luò)直接預(yù)測(cè)邊界框坐標(biāo)和該框內(nèi)的類。由于兩個(gè)階段融合在一起，所以一級(jí)檢測(cè)器往往比二級(jí)檢測(cè)器更快。但是由于兩個(gè)任務(wù)的分離，二級(jí)檢測(cè)器具有更高的精度。

3、分割

分割是計(jì)算機(jī)視覺(jué)中較獨(dú)特的任務(wù)之一，因?yàn)榫W(wǎng)絡(luò)既需要學(xué)習(xí)低級(jí)信息，也需要學(xué)習(xí)高級(jí)信息。低級(jí)信息可按像素精確分割圖像中的每個(gè)區(qū)域和對(duì)象，而高級(jí)信息可直接對(duì)這些像素進(jìn)行分類。這導(dǎo)致網(wǎng)絡(luò)被設(shè)計(jì)為將來(lái)自較早層和高分辨率（低層空間信息）的信息與較深層和低分辨率（高層語(yǔ)義信息）相結(jié)合。

如下所示，我們首先通過(guò)標(biāo)準(zhǔn)分類網(wǎng)絡(luò)運(yùn)行圖像。然后，我們從網(wǎng)絡(luò)的每個(gè)階段提取特征，從而使用從低到高的范圍內(nèi)的信息。每個(gè)信息級(jí)別在依次組合之前都是獨(dú)立處理的。當(dāng)這些信息組合在一起時(shí)，我們對(duì)特征圖進(jìn)行向上采樣，最終得到完整的圖像分辨率。

要了解更多關(guān)于如何分割與深度學(xué)習(xí)工作的細(xì)節(jié)，請(qǐng)查看這篇文章：
https://towardsdatascience.com/semantic-segmentation-with-deep-learning-...

4、姿態(tài)估計(jì)

姿態(tài)估計(jì)模型需要完成兩個(gè)任務(wù)：
（1）檢測(cè)圖像中每個(gè)身體部位的關(guān)鍵點(diǎn)；
（2）找出如何正確連接這些關(guān)鍵點(diǎn)。

這分以下三個(gè)階段完成：

① 使用標(biāo)準(zhǔn)分類網(wǎng)絡(luò)從圖像中提取特征。

② 給定這些特征，就可以訓(xùn)練一個(gè)子網(wǎng)絡(luò)來(lái)預(yù)測(cè)一組2D熱圖。每個(gè)熱圖都與一個(gè)特定的關(guān)鍵點(diǎn)相關(guān)聯(lián)，并包含每個(gè)圖像像素關(guān)于是否可能存在關(guān)鍵點(diǎn)的置信值。

③ 再次給出分類網(wǎng)絡(luò)的特征，我們訓(xùn)練一個(gè)子網(wǎng)絡(luò)來(lái)預(yù)測(cè)一組2D向量場(chǎng)，其中每個(gè)向量場(chǎng)都與關(guān)鍵點(diǎn)之間的關(guān)聯(lián)度進(jìn)行編碼。然后，具有較高關(guān)聯(lián)性的關(guān)鍵點(diǎn)被稱為已連接。

用這種方法訓(xùn)練子網(wǎng)絡(luò)的模型，可以聯(lián)合優(yōu)化關(guān)鍵點(diǎn)的檢測(cè)并將它們連接在一起。

5、增強(qiáng)和恢復(fù)

增強(qiáng)和恢復(fù)網(wǎng)絡(luò)是它們自己獨(dú)特的野獸。我們不會(huì)對(duì)此進(jìn)行任何向下采樣，因?yàn)槲覀冋嬲P(guān)心的是高像素/空間精度。向下采樣會(huì)真正抹殺這些信息，因?yàn)樗鼘p少我們?yōu)榭臻g精度而擁有的像素?cái)?shù)。相反，所有處理都是在全圖像分辨率下完成的。

我們開(kāi)始以全分辨率將想要增強(qiáng)/恢復(fù)的圖像傳遞到我們的網(wǎng)絡(luò)，而無(wú)需進(jìn)行任何修改。網(wǎng)絡(luò)僅由許多卷積和激活函數(shù)組成。這些塊通常是受啟發(fā)的，并且有時(shí)直接復(fù)制那些最初為圖像分類而開(kāi)發(fā)的塊，例如殘差塊、密集塊、擠壓激勵(lì)塊等。最后一層沒(méi)有激活函數(shù)，即使是sigmoid或softmax也沒(méi)有，因?yàn)槲覀兿胫苯宇A(yù)測(cè)圖像像素，不需要任何概率或分?jǐn)?shù)。

這就是所有這些類型的網(wǎng)絡(luò)。在圖像的全分辨率上進(jìn)行了大量的處理，來(lái)達(dá)到較高的空間精度，使用了與其他任務(wù)相同的卷積。

6、動(dòng)作識(shí)別

動(dòng)作識(shí)別是少數(shù)幾個(gè)需要視頻數(shù)據(jù)才能正常運(yùn)行的應(yīng)用程序之一。要對(duì)一個(gè)動(dòng)作進(jìn)行分類，我們需要了解隨著時(shí)間推移，場(chǎng)景中發(fā)生的變化， 這自然導(dǎo)致我們需要視頻。我們的網(wǎng)絡(luò)必須經(jīng)過(guò)訓(xùn)練來(lái)學(xué)習(xí)時(shí)空信息，即時(shí)空變化。最完美的網(wǎng)絡(luò)是3D-CNN。

顧名思義，3D-CNN是使用3D卷積的卷積網(wǎng)絡(luò)。它們與常規(guī)CNN的不同之處在于，卷積是在3維上應(yīng)用的：寬度、高度和時(shí)間。因此，每個(gè)輸出像素都是根據(jù)其周圍像素以及相同位置的前一幀和后一幀中的像素進(jìn)行計(jì)算來(lái)預(yù)測(cè)的。

視頻幀可以通過(guò)幾種方式傳遞：

直接在大批量中，例如第一個(gè)圖。由于我們正在傳遞一系列幀，因此空間和時(shí)間信息都是可用的。

我們還可以在一個(gè)流中傳遞單個(gè)圖像幀（數(shù)據(jù)的空間信息），并從視頻中傳遞其相應(yīng)的光流表示形式（數(shù)據(jù)的時(shí)間信息）。我們將使用常規(guī)2D CNN從這兩者中提取特征，然后再將其組合起來(lái)傳遞給我們的3D CNN，后者將兩種類型的信息進(jìn)行合并。

將幀序列傳遞給一個(gè)3D CNN，并將視頻的光流表示傳遞給另一個(gè)3D CNN。這兩個(gè)數(shù)據(jù)流都具有可用的空間和時(shí)間信息。鑒于我們正在對(duì)視頻的兩種不同表示（均包含我們的所有信息）進(jìn)行特定處理，因此這是最慢的選擇，但也可能是最準(zhǔn)確的選擇。

所有這些網(wǎng)絡(luò)都輸出視頻的動(dòng)作分類。

原文：https://towardsdatascience.com/how-to-do-everything-in-computer-vision-2...

文章來(lái)源于AI科技大本營(yíng)，作者George Seif