Zenn
😸

torchshow のメモ

2025/02/01に公開
visualization
tensorrt
idea

背景

Vision 系の ML をやっていて, テンソルや画像を可視化したい.

torchshow

軽く調べたかぎり, あんまり github に可視化ライブラリはないですが, その中でも一番機能ありそうなのとして torchshow がありました.

https://github.com/xwying/torchshow

修正

File ~/miniconda3/envs/qwen2vl/lib/python3.10/site-packages/torchshow/visualization.py:55, in set_window_title(fig, title)
  53 # fig.canvas.set_window_title(title)
     54 if matplotlib.__version__ < '3.4':
---> 55     fig.canvas.set_window_title(title)
     56 else:
     57     fig.canvas.manager.set_window_title(title)

AttributeError: 'FigureCanvasAgg' object has no attribute 'set_window_title'

というエラーがでる. バージョンを str で比較しているのが問題であろう.
コードを書き換えてあげれば治る
(執筆時点ではいる matplotlib は version 3.10 だった)

その他

https://github.com/apple/tensor-visualizer

開発は半年ほど止まっている.
最低限の可視化にはいいかもしれない

あとは tochshow なり参考にして, 自前で可視化してデバッグなどに活用していくのがよいであろう

Discussion