神经网络模型可视化工具、参数量、Flops 统计工具 |
您所在的位置：网站首页 › 神经网络flops › 神经网络模型可视化工具、参数量、Flops 统计工具 |
|
神经网络模型感受野计算工具 caffe模型可视化工具 netscope 左边放置.Prototxt 文本, 使用快捷键 shift + enter 可以绘制网络结构图 可视化工具 ConvNetDraw 利用pytorch 模型可视化工具pytorchviz Github 中的pytorchviz 安装 pip install graphviz pytorch使用tensorboard可视化参考教程 安装tensorboard conda install tensorboard 或 pip install tensorboard x = torch.rand([1, 3, 224, 640]) #.cuda() net = Yolov4tinysegfusion(11, 3, 2)# .cuda() with SummaryWriter(logdir='Yolov4tinysegfusion')as w: # tensorbnoard 模型可视化 w.add_graph(net,(x,))打开 terminal 输入 tensorboard --logdir=“logdir路径” TensorFlow installation not found - running with reduced feature set. Serving TensorBoard on localhost; to expose to the network, use a proxy or pass --bind_all TensorBoard 2.2.1 at http://localhost:6006/ (Press CTRL+C to quit) |
| CopyRight 2018-2019 办公设备维修网 版权所有 豫ICP备15022753号-3 |