訂閱
糾錯
加入自媒體

如何使用深度學(xué)習(xí)生成模糊背景?

步驟2:用于可視化從輸入中獲取的分割圖像的功能。def run_visualization():
 """Inferences DeepLab model and visualizes result."""
 try:
   original_im = Image.open(IMAGE_NAME)
 except IOError:
   print('Cannot retrieve image. Please check url: ' + url)
   returnprint('running deeplab on image')
 resized_im, seg_map = MODEL.run(original_im)
 vis_segmentation(resized_im, seg_map)
 return resized_im, seg_map

2.1:使用前面顯示的圖像調(diào)用上述功能。IMAGE_NAME = 'download2.jpg'
resized_im, seg_map = run_visualization()
分割后輸出。

2.2:現(xiàn)在,我們讀取輸入圖像并將其轉(zhuǎn)換為numpy數(shù)組。print(type(resized_im))
numpy_image = np.a(chǎn)rray(resized_im)

步驟3:分離背景和前景。在此步驟中,我們創(chuàng)建圖像的副本,然后,通過將背景中的值替換為0,并在已創(chuàng)建蒙版的位置保留255,將背景和前景與分割后的圖像分開,此處7表示汽車類別。person_not_person_mapping = deepcopy(numpy_image)
person_not_person_mapping[seg_map 。 7] = 0
person_not_person_mapping[seg_map == 7] = 255

3.1:可視化分離的蒙版圖像plt.imshow(person_not_person_mapping)

正如上一步中所述,背景已被黑色替換,汽車蒙版已變?yōu)榘咨瑯,通過替換這些值,我們也沒有丟失任何重要信息。

3.2:調(diào)整蒙版圖像的大小使其等于原始圖像。在分割過程之后,圖像的大小減小了,在我們的例子中,圖像的大小減小為(300 x 500),因此我們將圖像的大小調(diào)整為原始大小,即(900 x 596)。orig_imginal = Image.open(IMAGE_NAME)
orig_imginal = np.a(chǎn)rray(orig_imginal)mapping_resized = cv2.resize(person_not_person_mapping,
                            (orig_imginal.shape[1],
                             orig_imginal.shape[0]),
                             Image.ANTIALIAS)
mapping_resized.shape

3.3:二值化由于調(diào)整了大小,圖像生成的值在0,1,2…255之間,為了再次將值限制在0–255之間,我們必須使用Otsu的Binarization技術(shù)對圖像進行二值化。簡而言之,Otsu的Binarization是一種尋找灰度圖像閾值的自適應(yīng)方法,它遍歷0-255范圍內(nèi)的所有可能閾值,并找到給定圖像的最佳可能閾值。在內(nèi)部,它基于一些統(tǒng)計概念(例如方差),以根據(jù)所選閾值找出類別。一旦選擇了最佳閾值,則大于閾值的像素值將被視為白色像素,小于閾值的像素值將被視為黑色像素。

gray = cv2.cvtColor(mapping_resized, cv2.COLOR_BGR2GRAY)
blurred = cv2.GaussianBlur(gray,(15,15),0)
ret3,thresholded_img = cv2.threshold(blurred,0,255,cv2.THRESH_BINARY+cv2.THRESH_OTSU)
plt.imshow(thresholded_img)

<上一頁  1  2  3  4  下一頁>  余下全文
聲明: 本文由入駐維科號的作者撰寫,觀點僅代表作者本人,不代表OFweek立場。如有侵權(quán)或其他問題,請聯(lián)系舉報。

發(fā)表評論

0條評論,0人參與

請輸入評論內(nèi)容...

請輸入評論/評論長度6~500個字

您提交的評論過于頻繁,請輸入驗證碼繼續(xù)

暫無評論

暫無評論

    掃碼關(guān)注公眾號
    OFweek人工智能網(wǎng)
    獲取更多精彩內(nèi)容
    文章糾錯
    x
    *文字標(biāo)題:
    *糾錯內(nèi)容:
    聯(lián)系郵箱:
    *驗 證 碼:

    粵公網(wǎng)安備 44030502002758號