计算模型计算量MAC

onnx

python 复制代码
import onnx_tool

model_path = 'resnet50_224x224.onnx' #本地模型

onnx_tool.model_profile(model_path)

模型

python 复制代码
from torchprofile import profile_macs
from mirnet_v2_g3_conv13_noskff_arch import *
from thop import profile


if __name__ == '__main__':
    model_temp = MIRNet_v2(
        inp_channels=4,
        out_channels=4,
        n_feat=32,
        n_RRG=4,
        n_MRB=1
    )
    model_temp.eval()
    inp_frames = torch.randn(1, 4, 2568//2, 1440//2)
    #inp_frames = torch.randn(1, 4, 1920//2, 1080//2)
    inp_frames = torch.randn(1, 4, 128, 128)
    macs = profile_macs(model_temp, (inp_frames))
    print(macs // 1e9) # 计算量在 587 G 左右

    flops, params = profile(model_temp, inputs = (inp_frames, ))
    print('MACs = ' + str(flops/1000**3) + 'G' + '(PMRID)')
    print('Params = ' + str(params/1000**2) + 'M' )
相关推荐
*愿风载尘*4 小时前
SecureCRT连接密钥交换失败
linux·macos
mCell13 小时前
从删库到跑路?这50个Linux命令能保你职业生涯
linux·windows·macos
fhf21 小时前
2025年了你会卸载Macbook上的应用吗?
macos·shell
归辞...1 天前
「iOS」——内存五大分区
macos·ios·cocoa
穆雄雄1 天前
备份一下我的 mac mini 的环境变量配置情况
macos
秃然想通2 天前
mac电脑搭载c、c++环境(基于vs code)
macos
归辞...2 天前
「iOS」——GCD其他方法详解
macos·ios·cocoa
啊啊啊~~2 天前
新mac电脑软件安装指南(前端开发用)
macos·node·n·oh my zsh·solarized
lm down2 天前
ComfyUI中运行Wan 2.1工作流,电影级视频,兼容Mac, Windows
人工智能·macos·ai作画·视频
天安彩2 天前
mac下 vscode 运行 c++无法弹出窗口
c++·vscode·macos·clang