pytorch中的inference使用實(shí)例
更新時(shí)間：2020年02月20日 15:33:09 作者：Star_ACE
今天小編就為大家分享一篇pytorch中的inference使用實(shí)例，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過來看看吧
這里inference兩個(gè)程序的連接，如目標(biāo)檢測(cè)，可以利用一個(gè)程序提取候選框，然后把候選框輸入到分類cnn網(wǎng)絡(luò)中。
這里常需要進(jìn)行一定的連接。
#加載訓(xùn)練好的分類CNN網(wǎng)絡(luò)
model=torch.load('model.pkl')
#假設(shè)proposal_img是我們提取的候選框，是需要輸入到CNN網(wǎng)絡(luò)的數(shù)據(jù)
#先定義transforms對(duì)輸入cnn的網(wǎng)絡(luò)數(shù)據(jù)進(jìn)行處理，常包括resize、totensor等操作
data_transforms=transforms.Compose([transforms.RandomSizedCrop(224),
transforms.ToTensor()])
#由于transforms是對(duì)PIL格式數(shù)據(jù)操作，所以必要時(shí)轉(zhuǎn)化格式
def tensor_to_PIL(tensor):
image = tensor.cpu().clone()
image = image.squeeze(0)
image = unloader(image)
return image
#unqueeze（0）是加多一維，對(duì)應(yīng)原來batchsiaze
data=data_transforms(proposal_img).unqueeze(0)
#新版本pytorch已經(jīng)不用variable，可以省略這句
data=Variable（data）
#貌似這句也是多余的
torch.no_grad()
predict=F.softmax(model(data.cuda()).cuda())
以上這篇pytorch中的inference使用實(shí)例就是小編分享給大家的全部?jī)?nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。
相關(guān)文章
python自帶tkinter庫(kù)實(shí)現(xiàn)棋盤覆蓋圖形界面
這篇文章主要為大家詳細(xì)介紹了python自帶tkinter庫(kù)實(shí)現(xiàn)棋盤覆蓋圖形界面，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下2019-07-07
python 解決flask uwsgi 獲取不到全局變量的問題
今天小編就為大家分享一篇python 解決flask uwsgi 獲取不到全局變量的問題，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過來看看吧2019-12-12
pytorch 求網(wǎng)絡(luò)模型參數(shù)實(shí)例
今天小編就為大家分享一篇pytorch 求網(wǎng)絡(luò)模型參數(shù)實(shí)例，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過來看看吧2019-12-12
用?Python?腳本實(shí)現(xiàn)電腦喚醒后自動(dòng)拍照并截屏發(fā)郵件通知
這篇文章主要介紹了用?Python?腳本實(shí)現(xiàn)電腦喚醒后自動(dòng)拍照并截屏發(fā)郵件通知，文中詳細(xì)的介紹了代碼示例，具有一定的 參考價(jià)值，感興趣的可以了解一下2023-03-03
Python字符串、列表、元組、字典、集合的補(bǔ)充實(shí)例詳解
這篇文章主要介紹了Python字符串、列表、元組、字典、集合,結(jié)合實(shí)例形式詳細(xì)分析了Python字符串、列表、元組、字典、集合常見函數(shù)使用方法及相關(guān)操作注意事項(xiàng),需要的朋友可以參考下2019-12-12
這篇文章主要給大家介紹了關(guān)于pytorch?ssim計(jì)算的相關(guān)資料,結(jié)構(gòu)相似性(SSIM)是一種測(cè)量?jī)煞鶊D像的相似度的方法,文中通過代碼介紹的非常詳細(xì),需要的朋友可以參考下2023-12-12
分享8點(diǎn)超級(jí)有用的Python編程建議(推薦)
這篇文章主要介紹了分享8點(diǎn)超級(jí)有用的Python編程建議(推薦),小編覺得挺不錯(cuò)的，現(xiàn)在分享給大家，也給大家做個(gè)參考。一起跟隨小編過來看看吧2019-10-10
PyCharm 安裝與使用配置教程（windows,mac通用）
很多小伙伴下載安裝PyCharm后不會(huì)使用，這篇文章詳細(xì)介紹了PyCharm安裝與使用教程（windows,mac通用）,需要的朋友可以參考下2021-05-05
pycharm 將django中多個(gè)app放到同個(gè)文件夾apps的處理方法
今天小編就為大家分享一篇pycharm 將django中多個(gè)app放到同個(gè)文件夾apps的處理方法，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過來看看吧2018-05-05
Python中BaseHTTPRequestHandler實(shí)現(xiàn)簡(jiǎn)單的API接口
本文主要介紹了Python中BaseHTTPRequestHandler實(shí)現(xiàn)簡(jiǎn)單的API接口，文中通過示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2023-07-07