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