PyTorch 可視化工具:TensorBoard、Visdom

一筑辨、TensorBoard

TensorBoard 一般都是作為 TensorFlow 的可視化工具总滩,與 TensorFlow 深度集成纲堵,它能夠展現(xiàn) TensorFlow 的網(wǎng)絡(luò)計(jì)算圖,繪制圖像生成的定量指標(biāo)圖以及附加數(shù)據(jù)等闰渔。

此外席函,TensorBoard 也是一個(gè)獨(dú)立工具,在 PyTorch 中也可使用它進(jìn)行可視化冈涧。

1茂附、安裝:pip install tensorboard

2正蛙、啟動(dòng):tensorboard --logdir="日志目錄"

啟動(dòng) tensorboard 時(shí),可指定 logdir营曼、port(默認(rèn)6006)乒验、host(默認(rèn)localhost)等參數(shù):

usage: tensorboard [-h] [--helpfull] [--logdir PATH] [--logdir_spec PATH_SPEC]
                   [--host ADDR] [--bind_all] [--port PORT]
                   [--purge_orphaned_data BOOL] [--db URI] [--db_import]
                   [--inspect] [--version_tb] [--tag TAG] [--event_file PATH]
                   [--path_prefix PATH] [--window_title TEXT]
                   [--max_reload_threads COUNT] [--reload_interval SECONDS]
                   [--reload_task TYPE] [--reload_multifile BOOL]
                   [--reload_multifile_inactive_secs SECONDS]
                   [--generic_data TYPE]
                   [--samples_per_plugin SAMPLES_PER_PLUGIN]
                   [--debugger_data_server_grpc_port PORT]
                   [--debugger_port PORT] [--master_tpu_unsecure_channel ADDR]

3、Tensorboard 可視化演示(PyTorch 框架):

訓(xùn)練模型蒂阱,導(dǎo)入 tensorboard. SummaryWriter 保存 loss锻全、accuracy 等日志信息。

# 導(dǎo)入SummaryWriter
from torch.utils.tensorboard import SummaryWriter
...
# 創(chuàng)建SummaryWriter實(shí)例录煤,指定log_dir的位置
summaryWriter = SummaryWriter(log_dir="/Users/liyunfei/PycharmProjects/python3practice/06DL/fcnn/logs")
...
# 模型訓(xùn)練時(shí)鳄厌,寫(xiě)入train_loss、test_loss妈踊、score等信息
summaryWriter.add_scalars("loss", {"train_loss_avg": train_loss_avg, "test_loss_avg": test_loss_avg}, epoch)
summaryWriter.add_scalar("score", score, epoch)

啟動(dòng) TensorBoar 了嚎,訓(xùn)練過(guò)程可視化。
1)啟動(dòng)命令:

tensorboard --logdir=/Users/liyunfei/PycharmProjects/python3practice/06DL/fcnn/logs

2)啟動(dòng)成功如圖示:

3)可視化結(jié)果如下:

二廊营、Visdom

Visdom 是 Facebook 專(zhuān)門(mén)為 PyTorch 開(kāi)發(fā)的一款可視化工具新思,能夠支持“遠(yuǎn)程數(shù)據(jù)”的可視化,支持 Torch 和 Numpy赘风。GitHub 地址:https://github.com/fossasia/visdom

1夹囚、安裝:pip install visdom

2、啟動(dòng):

  • python -m visdom.server
    -m 是以模塊服務(wù)啟動(dòng)
  • 如果是 linux/mac-os 環(huán)境邀窃,可以使用以下命令啟動(dòng)運(yùn)行在后臺(tái)
    nohup python -m visdom.server &

啟動(dòng) Visdom 時(shí)荸哟,可以指定 port(默認(rèn)8097)、hostname(默認(rèn)localhost)等其它參數(shù):

usage: server.py [-h] [-port port] [--hostname hostname] [-base_url base_url]
                 [-env_path env_path] [-logging_level logger_level]
                 [-readonly] [-enable_login] [-force_new_cookie]
                 [-use_frontend_client_polling]

3瞬捕、Visdom 可視化演示

1)啟動(dòng) Visdom:

python -m visdom.server -port 8097

2)啟動(dòng)成功如下:

3)訓(xùn)練過(guò)程可視化代碼:

# 導(dǎo)入visdom包
import visdom

# 創(chuàng)建Visdom對(duì)象鞍历,連接服務(wù)端,指定env環(huán)境(不指定默認(rèn)env="main")
viz = visdom.Visdom(server='http://localhost', port=8097, env='liyunfei')
...
viz.line([0.], [0], win='train_loss', opts=dict(title='train_loss'))
viz.line([0.], [0], win='accuracy', opts=dict(title='accuracy'))
...
# 模型訓(xùn)練時(shí)肪虎,實(shí)時(shí)可視化loss劣砍、accuracy等信息。
viz.line([train_loss_avg], [epoch], win='train_loss', update='append')
viz.line([accuracy], [epoch], win='accuracy', update='append')

4)可視化結(jié)果:

5)其它操作——可視化一張/多張圖片:
示例:

import visdom
import numpy as np

viz = visdom.Visdom(server='http://localhost', port=8097, env='liyunfei')

# 一張圖片
viz.image(
    np.random.rand(3, 512, 256),
    opts=dict(title='Random!', caption='How random.'),
)
# 多張圖片
viz.images(
    np.random.randn(20, 3, 64, 64),
    nrow=5,
    opts=dict(title='Random images', caption='How random.')
)

效果:

6)Visdom 的更多可視化 API(常用的是 line扇救、image刑枝、text):

vis.scatter : 2D 或 3D 散點(diǎn)圖
vis.line : 線圖??
vis.stem : 莖葉圖
vis.heatmap : 熱力圖
vis.bar : 條形圖
vis.histogram: 直方圖
vis.boxplot : 箱型圖
vis.surf : 表面圖
vis.contour : 輪廓圖
vis.quiver : 繪出二維矢量場(chǎng)
vis.image : 圖片??
vis.text : 文本??
vis.mesh : 網(wǎng)格圖
vis.save : 序列化狀態(tài)
最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
  • 序言:七十年代末,一起剝皮案震驚了整個(gè)濱河市迅腔,隨后出現(xiàn)的幾起案子装畅,更是在濱河造成了極大的恐慌,老刑警劉巖沧烈,帶你破解...
    沈念sama閱讀 216,544評(píng)論 6 501
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件掠兄,死亡現(xiàn)場(chǎng)離奇詭異,居然都是意外死亡,警方通過(guò)查閱死者的電腦和手機(jī)蚂夕,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 92,430評(píng)論 3 392
  • 文/潘曉璐 我一進(jìn)店門(mén)迅诬,熙熙樓的掌柜王于貴愁眉苦臉地迎上來(lái),“玉大人婿牍,你說(shuō)我怎么就攤上這事侈贷。” “怎么了牍汹?”我有些...
    開(kāi)封第一講書(shū)人閱讀 162,764評(píng)論 0 353
  • 文/不壞的土叔 我叫張陵铐维,是天一觀的道長(zhǎng)。 經(jīng)常有香客問(wèn)我慎菲,道長(zhǎng)嫁蛇,這世上最難降的妖魔是什么? 我笑而不...
    開(kāi)封第一講書(shū)人閱讀 58,193評(píng)論 1 292
  • 正文 為了忘掉前任露该,我火速辦了婚禮睬棚,結(jié)果婚禮上,老公的妹妹穿的比我還像新娘解幼。我一直安慰自己抑党,他們只是感情好,可當(dāng)我...
    茶點(diǎn)故事閱讀 67,216評(píng)論 6 388
  • 文/花漫 我一把揭開(kāi)白布撵摆。 她就那樣靜靜地躺著底靠,像睡著了一般。 火紅的嫁衣襯著肌膚如雪特铝。 梳的紋絲不亂的頭發(fā)上暑中,一...
    開(kāi)封第一講書(shū)人閱讀 51,182評(píng)論 1 299
  • 那天,我揣著相機(jī)與錄音鲫剿,去河邊找鬼鳄逾。 笑死,一個(gè)胖子當(dāng)著我的面吹牛灵莲,可吹牛的內(nèi)容都是我干的雕凹。 我是一名探鬼主播,決...
    沈念sama閱讀 40,063評(píng)論 3 418
  • 文/蒼蘭香墨 我猛地睜開(kāi)眼政冻,長(zhǎng)吁一口氣:“原來(lái)是場(chǎng)噩夢(mèng)啊……” “哼枚抵!你這毒婦竟也來(lái)了?” 一聲冷哼從身側(cè)響起赠幕,我...
    開(kāi)封第一講書(shū)人閱讀 38,917評(píng)論 0 274
  • 序言:老撾萬(wàn)榮一對(duì)情侶失蹤俄精,失蹤者是張志新(化名)和其女友劉穎,沒(méi)想到半個(gè)月后榕堰,有當(dāng)?shù)厝嗽跇?shù)林里發(fā)現(xiàn)了一具尸體,經(jīng)...
    沈念sama閱讀 45,329評(píng)論 1 310
  • 正文 獨(dú)居荒郊野嶺守林人離奇死亡,尸身上長(zhǎng)有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點(diǎn)故事閱讀 37,543評(píng)論 2 332
  • 正文 我和宋清朗相戀三年逆屡,在試婚紗的時(shí)候發(fā)現(xiàn)自己被綠了圾旨。 大學(xué)時(shí)的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片。...
    茶點(diǎn)故事閱讀 39,722評(píng)論 1 348
  • 序言:一個(gè)原本活蹦亂跳的男人離奇死亡魏蔗,死狀恐怖砍的,靈堂內(nèi)的尸體忽然破棺而出,到底是詐尸還是另有隱情莺治,我是刑警寧澤廓鞠,帶...
    沈念sama閱讀 35,425評(píng)論 5 343
  • 正文 年R本政府宣布,位于F島的核電站谣旁,受9級(jí)特大地震影響床佳,放射性物質(zhì)發(fā)生泄漏。R本人自食惡果不足惜榄审,卻給世界環(huán)境...
    茶點(diǎn)故事閱讀 41,019評(píng)論 3 326
  • 文/蒙蒙 一砌们、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧搁进,春花似錦浪感、人聲如沸。這莊子的主人今日做“春日...
    開(kāi)封第一講書(shū)人閱讀 31,671評(píng)論 0 22
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽(yáng)。三九已至莱革,卻和暖如春峻堰,著一層夾襖步出監(jiān)牢的瞬間,已是汗流浹背驮吱。 一陣腳步聲響...
    開(kāi)封第一講書(shū)人閱讀 32,825評(píng)論 1 269
  • 我被黑心中介騙來(lái)泰國(guó)打工茧妒, 沒(méi)想到剛下飛機(jī)就差點(diǎn)兒被人妖公主榨干…… 1. 我叫王不留,地道東北人左冬。 一個(gè)月前我還...
    沈念sama閱讀 47,729評(píng)論 2 368
  • 正文 我出身青樓桐筏,卻偏偏與公主長(zhǎng)得像,于是被迫代替她去往敵國(guó)和親拇砰。 傳聞我的和親對(duì)象是個(gè)殘疾皇子梅忌,可洞房花燭夜當(dāng)晚...
    茶點(diǎn)故事閱讀 44,614評(píng)論 2 353

推薦閱讀更多精彩內(nèi)容