计算模型计算量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' )
相关推荐
博观而约取2 天前
Linux 和 macOS 终端中常见的快捷键操作
linux·运维·macos
Alger_Hamlet2 天前
Photoshop 2025 Mac中文 Ps图像编辑软件
macos·ui·photoshop
资源大全免费分享2 天前
MacOS 的 AI Agent 新星,本地沙盒驱动,解锁 macOS 操作新体验!
人工智能·macos·策略模式
刘小哈哈哈2 天前
封装了一个iOS多分区自适应宽度layout
macos·ios·cocoa
YJlio2 天前
TrollStore(巨魔商店)介绍及操作手册
macos·objective-c·cocoa
mywpython2 天前
mac 最新的chrome版本配置selenium的方式
chrome·python·selenium·macos
一道微光2 天前
mac air m系列arm架构芯片安装虚拟机 UTM+debian 浏览器firefox和chrome
arm开发·macos·架构
打工人你好2 天前
libimobiledevice项目中各个库的作用
macos·objective-c·cocoa
1alisa2 天前
Sublime Text for Mac v4【注册汉化版】代码编辑器
macos·编辑器·sublime text
qq_368019662 天前
Mac下Ollama安装与设置:开启本地大模型之旅
macos