计算FLOPs、Weights、特征可视化

连接参考
https://github.com/TingsongYu/ghostnet_cifar10

bin/03_compute_flops.py

前期准备:安装torchstat

Weights和FLOPs的计算采用torchstat工具,必须是python3以上才可安装

安装方法: pip install torchstat

torchstat网站:https://github.com/Swall0w/torchstat

方法:执行03_compute_flops.py 即可得到Weights和FLOPs

from torchstat import stat
import torchvision.models as models

model = models.resnet18()
stat(model, (3, 224, 224))

特征可视化
脚本:bin/04_fmap_vis.py

方法:运行脚本会在 results/runs下以时间戳为子文件夹记录下events file,然后借助tensorboard就可以查看特征图

你可能感兴趣的:(计算FLOPs、Weights、特征可视化)