快捷導(dǎo)航

Pytorch用Tensorboard來(lái)觀察數(shù)據(jù)

更新時(shí)間：2022年01月25日 12:57:34 作者：游客26024

這篇文章主要介紹了Pytorch用Tensorboard來(lái)觀察數(shù)據(jù)，上一篇文章我們講解了關(guān)于Pytorch?Dataset的數(shù)據(jù)處理，這篇我們就來(lái)講解觀察數(shù)據(jù)，下面具體相關(guān)資料,需要的朋友可以參考一下，希望對(duì)你有所幫助

1.Tensorboard

有兩個(gè)常用的方法：

一個(gè)是add_scalar()顯：示曲線
一個(gè)是add_image()顯示圖像

首先安裝Tensorboard

在你的編譯環(huán)境（conda activate XXX）中輸入命令

pip install tensorboard

1.使用add_scalar()輸入代碼

from torch.utils.tensorboard import SummaryWriter #調(diào)包

writer = SummaryWriter('logs') # 這里你創(chuàng)建了一個(gè)logs的文件裝你的add_scalar生成的曲線，
#其中writer.add_scalar（）第一個(gè)量是曲線的名字， 
#第二個(gè)量是縱坐標(biāo)scalar_value,第三個(gè)量是橫坐標(biāo)global_step（也可以理解為損失值得步長(zhǎng)）
for i in range(100):
    writer.add_scalar("quadratic", i ** 2, i)

writer.close()

打開(kāi)這個(gè)Tensorboard文件

tensorboard --logdir=logs
#1.這個(gè)logdir的文件名必須要與之前所創(chuàng)建的文件名一致，不然很容易報(bào)錯(cuò)，No dashboards are active for the current data set.
#2.這個(gè)tensorboard輸入的命令，必須是在logs文件的上一層文件中，不然也很容易報(bào)錯(cuò)，No dashboards are active for the current data set.

結(jié)果現(xiàn)實(shí)：

2.使用add_image()輸入代碼

注意：add_image()中函數(shù)一般有三個(gè)量:
第一個(gè)是圖像的名字，第二個(gè)是圖像(必須是tensor或者numpy.ndarray)，第三個(gè)是步長(zhǎng)（可理解為訓(xùn)練或者測(cè)試階段到哪幅圖像了）；
其中，圖像的shape必須是CHW，但是有opencv讀取的圖像shape是HWC,
所以得使用dataformats轉(zhuǎn)換以下將圖像的shape轉(zhuǎn)換為HWC

下面的代碼測(cè)試了兩張圖（一張是來(lái)自aligned的圖像，一張是來(lái)自original的圖像）用來(lái)模擬訓(xùn)練或者測(cè)試階段程序運(yùn)行到哪張圖

from torch.utils.tensorboard import SummaryWriter
import cv2

writer = SummaryWriter('logs')
aligned_img_path = "D:\\data\\basic\\Image\\aligned\\test_0001_aligned.jpg"
original_img_path = "D:\\data\\basic\\Image\\original\\test_0001.jpg"
aligned_img = cv2.imread(aligned_img_path)
original_img = cv2.imread(original_img_path)
print(type(aligned_img)) # numpy
print(aligned_img.shape)
# writer.add_image("img", aligned_img, 1, dataformats='HWC') #此圖已經(jīng)在我第一次測(cè)試add_image()用過(guò)了
writer.add_image("img", original_img, 2, dataformats='HWC')#此圖是我在第二個(gè)測(cè)試
writer.close()

實(shí)現(xiàn)結(jié)果：

tensorboard中出現(xiàn)了IMAGES，并且step1是aligned的圖，而step2是original的圖

到此這篇關(guān)于Pytorch用Tensorboard來(lái)觀察數(shù)據(jù)的文章就介紹到這了,更多相關(guān)Pytorch使用Tensorboard內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: