圖像分割中原圖和mask結(jié)合

import cv2
import os
import numpy as np

def union_image_mask(image_path, mask_path, image_name, color = (0, 255, 0)):
    image = cv2.imread(image_path)
    chang = image.shape[1]
    kuan = image.shape[0]
    mask_2d = cv2.imread(mask_path,0)
    mask_2d = cv2.resize(mask_2d,(chang, kuan))

    coef = 255 if np.max(image)<3 else 1
    image = (image * coef).astype(np.float32)
    contours, _ = cv2.findContours(mask_2d, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)

    print(image.shape)
    cv2.drawContours(image, contours, -1, color, 1)
    cv2.imwrite(os.path.join('Add', image_name),image)



def change_image(path):
    for img_name in os.listdir(path):
        print(img_name)
        img_path = os.path.join(path, img_name)
        img = cv2.imread(img_path)

        img = cv2.cvtColor(img, cv2.COLOR_RGB2GRAY)
        img = cv2.cvtColor(img, cv2.COLOR_GRAY2RGB)
        img[img >0 ] = 255
        cv2.imwrite(img_path, img)


if __name__ == '__main__':
    images = os.listdir('dataset/11') # 原圖文件夾
    mask = os.listdir('output') # mask文件夾
    for image_name in images:
        image_path = os.path.join('dataset/11', image_name)
        mask_path = os.path.join('output', image_name)
        union_image_mask(image_path, mask_path,image_name)
    #change_image('output')
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
  • 序言:七十年代末杯巨,一起剝皮案震驚了整個(gè)濱河市求豫,隨后出現(xiàn)的幾起案子,更是在濱河造成了極大的恐慌寸宵,老刑警劉巖在孝,帶你破解...
    沈念sama閱讀 211,194評(píng)論 6 490
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件,死亡現(xiàn)場(chǎng)離奇詭異钟鸵,居然都是意外死亡性昭,警方通過查閱死者的電腦和手機(jī)渣蜗,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 90,058評(píng)論 2 385
  • 文/潘曉璐 我一進(jìn)店門屠尊,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人袍睡,你說我怎么就攤上這事知染。” “怎么了斑胜?”我有些...
    開封第一講書人閱讀 156,780評(píng)論 0 346
  • 文/不壞的土叔 我叫張陵,是天一觀的道長(zhǎng)嫌吠。 經(jīng)常有香客問我止潘,道長(zhǎng),這世上最難降的妖魔是什么辫诅? 我笑而不...
    開封第一講書人閱讀 56,388評(píng)論 1 283
  • 正文 為了忘掉前任凭戴,我火速辦了婚禮,結(jié)果婚禮上炕矮,老公的妹妹穿的比我還像新娘么夫。我一直安慰自己,他們只是感情好肤视,可當(dāng)我...
    茶點(diǎn)故事閱讀 65,430評(píng)論 5 384
  • 文/花漫 我一把揭開白布档痪。 她就那樣靜靜地躺著,像睡著了一般邢滑。 火紅的嫁衣襯著肌膚如雪腐螟。 梳的紋絲不亂的頭發(fā)上,一...
    開封第一講書人閱讀 49,764評(píng)論 1 290
  • 那天困后,我揣著相機(jī)與錄音乐纸,去河邊找鬼。 笑死摇予,一個(gè)胖子當(dāng)著我的面吹牛汽绢,可吹牛的內(nèi)容都是我干的。 我是一名探鬼主播侧戴,決...
    沈念sama閱讀 38,907評(píng)論 3 406
  • 文/蒼蘭香墨 我猛地睜開眼宁昭,長(zhǎng)吁一口氣:“原來是場(chǎng)噩夢(mèng)啊……” “哼!你這毒婦竟也來了救鲤?” 一聲冷哼從身側(cè)響起久窟,我...
    開封第一講書人閱讀 37,679評(píng)論 0 266
  • 序言:老撾萬(wàn)榮一對(duì)情侶失蹤,失蹤者是張志新(化名)和其女友劉穎本缠,沒想到半個(gè)月后斥扛,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體,經(jīng)...
    沈念sama閱讀 44,122評(píng)論 1 303
  • 正文 獨(dú)居荒郊野嶺守林人離奇死亡,尸身上長(zhǎng)有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點(diǎn)故事閱讀 36,459評(píng)論 2 325
  • 正文 我和宋清朗相戀三年稀颁,在試婚紗的時(shí)候發(fā)現(xiàn)自己被綠了芬失。 大學(xué)時(shí)的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片。...
    茶點(diǎn)故事閱讀 38,605評(píng)論 1 340
  • 序言:一個(gè)原本活蹦亂跳的男人離奇死亡匾灶,死狀恐怖棱烂,靈堂內(nèi)的尸體忽然破棺而出,到底是詐尸還是另有隱情阶女,我是刑警寧澤颊糜,帶...
    沈念sama閱讀 34,270評(píng)論 4 329
  • 正文 年R本政府宣布,位于F島的核電站秃踩,受9級(jí)特大地震影響衬鱼,放射性物質(zhì)發(fā)生泄漏。R本人自食惡果不足惜憔杨,卻給世界環(huán)境...
    茶點(diǎn)故事閱讀 39,867評(píng)論 3 312
  • 文/蒙蒙 一鸟赫、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧消别,春花似錦抛蚤、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 30,734評(píng)論 0 21
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽(yáng)。三九已至荆虱,卻和暖如春蒿偎,著一層夾襖步出監(jiān)牢的瞬間,已是汗流浹背怀读。 一陣腳步聲響...
    開封第一講書人閱讀 31,961評(píng)論 1 265
  • 我被黑心中介騙來泰國(guó)打工诉位, 沒想到剛下飛機(jī)就差點(diǎn)兒被人妖公主榨干…… 1. 我叫王不留,地道東北人菜枷。 一個(gè)月前我還...
    沈念sama閱讀 46,297評(píng)論 2 360
  • 正文 我出身青樓苍糠,卻偏偏與公主長(zhǎng)得像,于是被迫代替她去往敵國(guó)和親啤誊。 傳聞我的和親對(duì)象是個(gè)殘疾皇子岳瞭,可洞房花燭夜當(dāng)晚...
    茶點(diǎn)故事閱讀 43,472評(píng)論 2 348