pytorch獲取vgg16-feature層輸出的例子
實際應用時可能比較想獲取VGG中間層的輸出,
那么就可以如下操作:
import numpy as np import torch from torchvision import models from torch.autograd import Variable import torchvision.transforms as transforms class CNNShow(): def __init__(self, model): self.model = model self.model.eval() self.created_image = self.image_for_pytorch(np.uint8(np.random.uniform(150, 180, (224, 224, 3)))) def show(self): x = self.created_image for index, layer in enumerate(self.model): print(index,layer) x = layer(x) def image_for_pytorch(self,Data): transform = transforms.Compose([ transforms.ToTensor(), # range [0, 255] -> [0.0,1.0] transforms.Normalize(mean=(0.485, 0.456, 0.406), std=(0.229, 0.224, 0.225)) ] ) imData = transform(Data) imData = Variable(torch.unsqueeze(imData, dim=0), requires_grad=True) return imData if __name__ == '__main__': pretrained_model = models.vgg16(pretrained=True).features CNN = CNNShow(pretrained_model) CNN.show()
以上這篇pytorch獲取vgg16-feature層輸出的例子就是小編分享給大家的全部內(nèi)容了,希望能給大家一個參考,也希望大家多多支持腳本之家。
相關文章
使用python3調(diào)用wxpy模塊監(jiān)控linux日志并定時發(fā)送消息給群組或好友
這篇文章主要介紹了使用python3調(diào)用wxpy模塊,監(jiān)控linux日志并定時發(fā)送消息給群組或好友,本文通過實例代碼給大家介紹的非常詳細,具有一定的參考借鑒價值,需要的朋友可以參考下2019-06-06Python多任務版靜態(tài)Web服務器實現(xiàn)示例
這篇文章主要為大家介紹了Python靜態(tài)Web服務器多任務版實現(xiàn)示例,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進步,早日升職加薪2022-06-06解決python調(diào)用自己文件函數(shù)/執(zhí)行函數(shù)找不到包問題
這篇文章主要介紹了解決python調(diào)用自己文件函數(shù)/執(zhí)行函數(shù)找不到包問題,具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧2020-06-06解決numpy矩陣相減出現(xiàn)的負值自動轉(zhuǎn)正值的問題
這篇文章主要介紹了解決numpy矩陣相減出現(xiàn)的負值自動轉(zhuǎn)正值的問題,具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧2020-06-06python數(shù)據(jù)可視化 – 利用Bokeh和Bottle.py在網(wǎng)頁上展示你的數(shù)據(jù)
本文將展示如何使用python搭建一個網(wǎng)頁應用來展示你的數(shù)據(jù)圖表 很多有關于使用python搭建網(wǎng)頁應用的文章聚焦在如何教讀者搭建一個網(wǎng)頁應用(大多是博客),很多關于使用python做數(shù)據(jù)可視化的文章聚焦在如何教讀者使用python的圖表庫來做可視化2021-10-10win10系統(tǒng)下python3安裝及pip換源和使用教程
這篇文章主要介紹了win10系統(tǒng)下python3安裝及pip換源和使用教程,本文給大家介紹的非常詳細,具有一定的參考借鑒價值,需要的朋友可以參考下2020-01-01