pytorch獲取vgg16-feature層輸出的例子
實(shí)際應(yīng)用時(shí)可能比較想獲取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層輸出的例子就是小編分享給大家的全部?jī)?nèi)容了,希望能給大家一個(gè)參考,也希望大家多多支持腳本之家。
相關(guān)文章
使用python3調(diào)用wxpy模塊監(jiān)控linux日志并定時(shí)發(fā)送消息給群組或好友
這篇文章主要介紹了使用python3調(diào)用wxpy模塊,監(jiān)控linux日志并定時(shí)發(fā)送消息給群組或好友,本文通過(guò)實(shí)例代碼給大家介紹的非常詳細(xì),具有一定的參考借鑒價(jià)值,需要的朋友可以參考下2019-06-06Python多任務(wù)版靜態(tài)Web服務(wù)器實(shí)現(xiàn)示例
這篇文章主要為大家介紹了Python靜態(tài)Web服務(wù)器多任務(wù)版實(shí)現(xiàn)示例,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進(jìn)步,早日升職加薪2022-06-06利用PyQt中的QThread類(lèi)實(shí)現(xiàn)多線程
本文主要給大家分享的是python實(shí)現(xiàn)多線程及線程間通信的簡(jiǎn)單方法,非常的實(shí)用,有需要的小伙伴可以參考下2020-02-02解決python調(diào)用自己文件函數(shù)/執(zhí)行函數(shù)找不到包問(wèn)題
這篇文章主要介紹了解決python調(diào)用自己文件函數(shù)/執(zhí)行函數(shù)找不到包問(wèn)題,具有很好的參考價(jià)值,希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2020-06-06Python中for循環(huán)控制語(yǔ)句用法實(shí)例
這篇文章主要介紹了Python中for循環(huán)控制語(yǔ)句用法,較為詳細(xì)的分析了for循環(huán)語(yǔ)句的原理與相關(guān)使用技巧,需要的朋友可以參考下2015-06-06解決numpy矩陣相減出現(xiàn)的負(fù)值自動(dòng)轉(zhuǎn)正值的問(wèn)題
這篇文章主要介紹了解決numpy矩陣相減出現(xiàn)的負(fù)值自動(dòng)轉(zhuǎn)正值的問(wèn)題,具有很好的參考價(jià)值,希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2020-06-06Python實(shí)現(xiàn)抓取頁(yè)面上鏈接的簡(jiǎn)單爬蟲(chóng)分享
這篇文章主要介紹了Python實(shí)現(xiàn)抓取頁(yè)面上鏈接的簡(jiǎn)單爬蟲(chóng)分享,本文使用了一個(gè)開(kāi)源模塊requests實(shí)現(xiàn)需求,需要的朋友可以參考下2015-01-01python數(shù)據(jù)可視化 – 利用Bokeh和Bottle.py在網(wǎng)頁(yè)上展示你的數(shù)據(jù)
本文將展示如何使用python搭建一個(gè)網(wǎng)頁(yè)應(yīng)用來(lái)展示你的數(shù)據(jù)圖表 很多有關(guān)于使用python搭建網(wǎng)頁(yè)應(yīng)用的文章聚焦在如何教讀者搭建一個(gè)網(wǎng)頁(yè)應(yīng)用(大多是博客),很多關(guān)于使用python做數(shù)據(jù)可視化的文章聚焦在如何教讀者使用python的圖表庫(kù)來(lái)做可視化2021-10-10win10系統(tǒng)下python3安裝及pip換源和使用教程
這篇文章主要介紹了win10系統(tǒng)下python3安裝及pip換源和使用教程,本文給大家介紹的非常詳細(xì),具有一定的參考借鑒價(jià)值,需要的朋友可以參考下2020-01-01