Tensorboard 可視化工具的使用-史上最簡單(附代碼)

一.Tensorboard是什么? 為什么使用? TensorboardfX是什么?

在深度學習中,我們需要對定義的網絡訓練的效果進行了解,但是通過代碼不是特別的直觀,這時候我們就需要借助一些可視化工具來幫助我們查看訓練的結果.

• 是什么:

  tensorboard是一種用來可視化的工具

• 為什么使用?

  可以監督網絡的訓練過程或呈現網絡的訓練結果,以獲得更有效的訓練結果

• TensorboardX:

  tensorboardX是幫助pytorch使用tensorboard工具來可視化的一個庫,其中包含若干函數用於簡化tensorboard的使用

---

二.tensorboardX常用的函數介紹?

1. SummaryWriter()
   • 功能:創建編寫器,保存日志
   • 用法:writer = SummaryWriter()
2. writer.add_scalar()
   • 功能:添加標量
   • 用法:writer.add_scalar('標題','y軸','x軸')
3. writer.add_image()
   • 功能:添加圖像
   • 用法:writer.add_image('標題','tensor類型的數據','同一標題下的步驟')

---

三.如何使用tensorboard 及 打開tensorboard界面

1. 首先引用以下的模塊

from tensorboardX import SummaryWriter

2. 下面我們以添加標量為例來說明

• 第一步

  中間是保存的地址

  writer = SummaryWriter('LOGS/log_tensorboardX')
  

• 第二步 使用add_scalar函數來向log_tensorboardX中寫入標量

  for i in range(100):
      writer.add_scalar('y=x', i, i)
  

• 第三步

  關閉工作流

  writer.close()
  

3. 打開tensorboard可視化界面

在終端中輸入以下的指令,會出現一個網址,點擊即可打開tensorboard界面

​

---

四.小結

以上只舉例了writer_scalar函數如何使用,關於后續函數,如果我學習使用到了就會再次來記錄.

每天一句:制心一處,無事不辦