计算模型计算量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' )
相关推荐
Dachui_112210 小时前
MacOS上如何运行内网穿透详细教程
macos
Bruce_Liuxiaowei12 小时前
使用Python脚本在Mac上彻底清除Chrome浏览历史:开发实战与隐私保护指南
chrome·python·macos
iFlyCai14 小时前
对Mac文字双击或三击鼠标左键没有任何反应
macos·计算机外设
請你喝杯Java1 天前
Mac上Cursor无法安装插件解决方法
macos·cursor·vsode
网络之路Blog1 天前
【实战中提升自己】内网安全部署之端口隔离与MAC地址认证
安全·macos·网络之路一天·华为华三数通基础·华为华三企业实战架构·华为中小型企业实战·华为华三计算机网络基础
心随_风动1 天前
主流操作系统对比分析(macOS、Linux、Windows、Unix)
linux·windows·macos
lxw18449125141 天前
macOS 系统设置息屏情况下,PHP等后台脚本继续执行
macos
Brian_Lucky1 天前
在 macOS 上合并 IntelliJ IDEA 的项目窗口
java·macos·intellij-idea
未来之窗软件服务1 天前
声音分离人声和配乐-从头设计数字生命第5课, demucs——仙盟创梦IDE
ide·macos·xcode·仙盟创梦ide·数字生命
我该如何取个名字2 天前
Mac mini 安装mysql数据库以及出现的一些问题的解决方案
数据库·mysql·macos