pytorch 模型視覺化

針對pytorch有很多視覺化方式，比如，tensorboard_logger、visdom等，但這些比較複雜和不夠友好，則使用更強大的工具--tenorboardx

from torch.utils.tensorboard import summarywriter
if __name__ == "__main__":
input = torch.tensor(2, 3, 112, 112)
net = seresnet34_ir()
with summarywriter(log_dir='logs', comment='net') as w:
w.add_graph(net, (input, ))

# cd 到logs上級目錄上，在win10上cmd的pytorch環境下輸入

tensorboard --logdir=logs --port 6066

更新【2023年8月27】

怎麼都顯示不出來，

其實是在什麼pytorch環境中生成logs的，就需要在該環境下開啟。

我在pytorch=1.2中生成的比在1.5的少個幾十kb，所以在1.5環境下生成graph，然後在該環境下使用：

# cd 到logs上級目錄上，在win10上cmd的pytorch環境下輸入

tensorboard --logdir=logs --port 6066

完美出來

pytorch模型視覺化

先把tensorflow和pytorch安裝好 之後在pytorch環境裡 pip install tensorboardx 將with summarywriter comment lenet as w w.add graph model,varinput,新增到模型和輸入的後面，類似這樣注意引數第...

PyTorch 視覺化特徵

這個也可以參考 這篇部落格主要記錄了如何提取特定層的特徵，然後對它進行視覺化 處理單張作為網路輸入。根據給定的layer層，獲取該層的輸出結果features。考慮到features的形狀為 batch size,filter nums,h,w 提取其中的第乙個過濾器得到的結果feature。以一張...

pytorch視覺化方法

netron 二.netron 示例 pandas 是基於numpy 的一種工具，該工具是為了解決資料分析任務而建立的。sudo apt install graphviz pip install graphviz pip install tochviz 或pip install git https ...