计算模型计算量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' )
相关推荐
梅名智15 小时前
IOS 蓝牙连接
macos·ios·cocoa
帅次1 天前
Objective-C面向对象编程:类、对象、方法详解(保姆级教程)
flutter·macos·ios·objective-c·iphone·swift·safari
fengyun28911 天前
fcpx音视频剪辑编辑 Final Cut Pro X(Mac电脑)
macos·mac
fengyun28911 天前
Mac电脑 卸载工具 App Cleaner
macos
aerror1 天前
Xbox One 控制器转换为 macOS HID 设备的工作原理分析
macos·xbox
晴风向上1 天前
mac mini m4安装node.js@16以下版本方法
macos·node.js
JPCstorm2 天前
Mac homebrew 安装教程
macos
秋窗72 天前
Mac 部署Latex OCR并优化体验(打包成App并支持全局快捷键)
macos·ocr·latex
FreeBuf_2 天前
朝鲜APT组织使用Nim语言恶意软件对macOS发起隐秘Web3与加密货币攻击
macos·web3·策略模式
Digitally2 天前
如何将信息从 iPhone 同步到Mac(完整步骤和示意图)
macos·ios·iphone