日韩黑丝制服一区视频播放|日韩欧美人妻丝袜视频在线观看|九九影院一级蜜桃|亚洲中文在线导航|青草草视频在线观看|婷婷五月色伊人网站|日本一区二区在线|国产AV一二三四区毛片|正在播放久草视频|亚洲色图精品一区

分享

keras加載h5模型文件,進(jìn)行圖像分類(lèi)任務(wù)的結(jié)果預(yù)測(cè)。

 行走在理想邊緣 2022-05-16 發(fā)布于四川

這是一個(gè)皮膚病8分類(lèi)任務(wù)。首先展示類(lèi)別名稱(chēng)列表:

classes_name_list=['AK','BCC','BKL','DF','MEL','NV','SCC','VASC']

以上代碼表明,分類(lèi)任務(wù)為8分類(lèi),標(biāo)簽分別為AK,BCC,BKL等等。

通過(guò)對(duì)DenseNet121進(jìn)行遷移學(xué)習(xí),將頂層修改為自己設(shè)計(jì)的8分類(lèi)器,并在訓(xùn)練集上訓(xùn)練,(訓(xùn)練圖像全部歸一化為0-1范圍內(nèi)),獲得訓(xùn)練好的模型:model.h5

加載該預(yù)訓(xùn)練模型:

import tensorflow as tfmodel=tf.keras.models.load_model('model.h5')

該模型結(jié)構(gòu)如下圖:

1.首先我們使用skimage模塊來(lái)預(yù)處理圖像:

加載待預(yù)測(cè)圖像(皮膚病圖像辣眼睛):

import skimagefilename=r'datasets\test\MEL\ISIC_0000013.jpg'test_img=skimage.io.imread(filename) #加載測(cè)試圖像,直接得到多維數(shù)組test_img=skimage.transform.resize(test_img,(224,224,3))  #根據(jù)模型輸入尺寸要求,重定義大小test_img=np.expand_dims(test_img,0) #模型輸入要求為四維張量,所以需要增加一個(gè)batch_size維度pred=model.predict(test_img) #將測(cè)試圖像送入模型,進(jìn)行預(yù)測(cè)print('預(yù)測(cè)結(jié)果:',end='')print(classes_name_list[pred.argmax()]) #輸出類(lèi)別預(yù)測(cè)結(jié)果
預(yù)測(cè)結(jié)果:MEL

2.接著換用keras自帶的圖像預(yù)處理模塊進(jìn)行預(yù)處理:

import tensorflow as tffilename=r'datasets\test\MEL\ISIC_0000013.jpg'test_img=tf.keras.preprocessing.image.load_img(filename,target_size=(224,224,3)) #此處得到的是pillow圖像Image實(shí)例test_img=tf.keras.preprocessing.image.img_to_array(test_img) #將Image實(shí)例轉(zhuǎn)化為多維數(shù)組test_img=test_img/255  #此處還需要將0-255轉(zhuǎn)化為0-1test_img=np.expand_dims(test_img,0) #將三維輸入圖像拓展成四維張量pred=model.predict(test_img) #預(yù)測(cè)print('預(yù)測(cè)結(jié)果:',end='')print(classes_name_list[pred.argmax()])  #打印結(jié)果
預(yù)測(cè)結(jié)果:MEL

可以看出,使用skimage和keras自帶的圖像預(yù)處理模塊都能實(shí)現(xiàn)準(zhǔn)確預(yù)測(cè),唯一不同是,skimage讀取圖像后自動(dòng)轉(zhuǎn)換為0-1范圍,而keras中使用的需要手動(dòng)轉(zhuǎn)換0-255為0-1.

    本站是提供個(gè)人知識(shí)管理的網(wǎng)絡(luò)存儲(chǔ)空間,所有內(nèi)容均由用戶發(fā)布,不代表本站觀點(diǎn)。請(qǐng)注意甄別內(nèi)容中的聯(lián)系方式、誘導(dǎo)購(gòu)買(mǎi)等信息,謹(jǐn)防詐騙。如發(fā)現(xiàn)有害或侵權(quán)內(nèi)容,請(qǐng)點(diǎn)擊一鍵舉報(bào)。
    轉(zhuǎn)藏 分享 獻(xiàn)花(0

    0條評(píng)論

    發(fā)表

    請(qǐng)遵守用戶 評(píng)論公約

    類(lèi)似文章 更多