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ù)操作，所以必要時轉(zhuǎn)化格式
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)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關(guān)文章
解決python遞歸函數(shù)及遞歸次數(shù)受到限制的問題
這篇文章主要介紹了解決python遞歸函數(shù)及遞歸次數(shù)受到限制的問題，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教2022-06-06
這篇文章主要介紹了Python pymongo模塊常用操作,結(jié)合實例形式分析了pymongo模塊的導入、連接MongoDB數(shù)據(jù)以及插入、更新、刪除、查詢等操作相關(guān)實現(xiàn)技巧,需要的朋友可以參考下2018-09-09
利用Python操作MongoDB數(shù)據(jù)庫的詳細指南
MongoDB是由C++語言編寫的非關(guān)系型數(shù)據(jù)庫,是一個基于分布式文件存儲的開源數(shù)據(jù)庫系統(tǒng),其內(nèi)容存儲形式類似JSON對象,下面這篇文章主要給大家介紹了關(guān)于利用Python操作MongoDB數(shù)據(jù)庫的相關(guān)資料,需要的朋友可以參考下2023-02-02
這篇文章主要介紹了python實現(xiàn)的B站直播錄播工具，幫助大家更好的理解和學習使用python，感興趣的朋友可以了解下2021-04-04
python針對mysql數(shù)據(jù)庫的連接、查詢、更新、刪除操作示例
這篇文章主要介紹了python針對mysql數(shù)據(jù)庫的連接、查詢、更新、刪除操作,結(jié)合實例形式詳細分析了Python操作mysql數(shù)據(jù)庫的連接與增刪改查相關(guān)實現(xiàn)技巧,需要的朋友可以參考下2019-09-09
這篇文章主要為大家介紹了Python編程獲取終端命令行參數(shù)示例，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進步，早日升職加薪2022-06-06
Python調(diào)用pyttsx3實現(xiàn)離線文字轉(zhuǎn)語音的方式
pyttsx3是 Python 中的文本到語音的離線轉(zhuǎn)換庫,本文給大家介紹Python調(diào)用pyttsx3實現(xiàn)離線文字轉(zhuǎn)語音的方式,感興趣的朋友一起看看吧2024-03-03
這篇文章主要介紹了Python變量的賦值、淺拷貝和深拷貝詳解,python中為聲明一個變量有三種方法：賦值、淺拷貝、深拷貝,相信每個pythoner或多或少都知道他們之間的區(qū)別,但在某些點上,還是會踩坑,這篇文章記錄下所有關(guān)于這三者區(qū)別的疑問,需要的朋友可以參考下2023-11-11
python中判斷數(shù)字是否為質(zhì)數(shù)的實例講解
在本篇文章里小編給大家分享了關(guān)于python中判斷數(shù)字是否為質(zhì)數(shù)的實例講解內(nèi)容，有興趣的朋友們可以學習下。2020-12-12
這篇文章主要介紹了Python如何使用paramiko模塊連接linux,文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值,需要的朋友可以參考下2020-03-03