模型可视化工具和库

参考:Pytorch查看网络结构

小结模型可视化工具和库

Thop

仓库链接:Lyken17/pytorch-OpCounter

Thop能够计算模型参数数目以及Flops。其具体实现参考参数数目和Flops计算

Netscope CNN Analyzer

参考:how to calculate a net’s FLOPs in CNN

Netscope CNN Analyzer是一个在线可视化网络模型网站,支持Caffeprototxt格式,也提供了很多模型的实现

pytorchviz

仓库链接:szagoruyko/pytorchviz

pytorchviz能够生成PyTorch模型的执行图

torchsummary

仓库链接:sksq96/pytorch-summary

torchsummary能够输出模型每一层的参数数目

坚持原创技术分享,您的支持将鼓励我继续创作!