pytorch 歸一化與反歸一化實(shí)例
更新時間:2019年12月31日 15:35:39 作者:ShellCollector
今天小編就為大家分享一篇pytorch 歸一化與反歸一化實(shí)例,具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧
ToTensor中就有轉(zhuǎn)到0-1之間了。
# -*- coding:utf-8 -*-
import time
import torch
from torchvision import transforms
import cv2
transform_val_list = [
# transforms.Resize(size=(160, 160), interpolation=3), # Image.BICUBIC
transforms.ToTensor(),
transforms.Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225])
]
trans_compose = transforms.Compose(transform_val_list)
if __name__ == '__main__':
std= [0.229, 0.224, 0.225]
mean=[0.485, 0.456, 0.406]
path="d:/2.jpg"
data=cv2.imread(path)
t1 = time.time()
x = trans_compose(data)
x[0]=x[0]*std[0]+mean[0]
x[1]=x[1]*std[1]+mean[1]
x[2]=x[2].mul(std[2])+mean[2]
img = x.mul(255).byte()
img = img.numpy().transpose((1, 2, 0))
# torch.set_num_threads(3)
# img=cv2.cvtColor(img,cv2.COLOR_BGR2RGB)
cv2.imshow("sdf", img)
cv2.waitKeyEx()
這個測試時間:歸一化與反歸一化都需要7ms左右,
但是在多路攝像頭中,可能比較慢。
std= [0.229, 0.224, 0.225]
mean=[0.485, 0.456, 0.406]
path="d:/2.jpg"
data=cv2.imread(path)
t1 = time.time()
start = time.time()
x = trans_compose(data)
print("gui", time.time() - start)
for i in range(10):
start=time.time()
for i in range(len(mean)):
# x[i]=x[i]*std[i]+mean[i]
x[i]=x[i].mul(std[i])+mean[i]
img = x.mul(255).byte()
img = img.numpy().transpose((1, 2, 0))
print("fan",time.time()-start)
# torch.set_num_threads(3)
# img=cv2.cvtColor(img,cv2.COLOR_BGR2RGB)
cv2.imshow("sdf", img)
cv2.waitKeyEx()
以上這篇pytorch 歸一化與反歸一化實(shí)例就是小編分享給大家的全部內(nèi)容了,希望能給大家一個參考,也希望大家多多支持腳本之家。
相關(guān)文章
解決pycharm回車之后不能換行或不能縮進(jìn)的問題
今天小編就為大家分享一篇解決pycharm回車之后不能換行或不能縮進(jìn)的問題,具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧2019-01-01
OpenCV-Python實(shí)現(xiàn)圖像平滑處理操作
圖像平滑處理的噪聲取值主要有6種方法,本文主要介紹了這6種方法的具體使用并配置實(shí)例方法,具有一定的參考價值,感興趣的可以了解一下2021-06-06
python獲取天氣接口給指定微信好友發(fā)天氣預(yù)報
這篇文章主要介紹了python獲取天氣接口給指定微信好友發(fā)天氣預(yù)報的步驟,幫助大家更好的理解和學(xué)習(xí)python,感興趣的朋友可以了解下2020-12-12
python實(shí)現(xiàn)簡單聊天應(yīng)用 python群聊和點(diǎn)對點(diǎn)均實(shí)現(xiàn)
這篇文章主要為大家詳細(xì)介紹了python實(shí)現(xiàn)簡單聊天應(yīng)用,python群聊和點(diǎn)對點(diǎn)均實(shí)現(xiàn),具有一定的參考價值,感興趣的小伙伴們可以參考一下2017-09-09

