RT-DETR 一键输出FPS、层数、参数量、计算量、模型大小、推理延时

前言

在官方项目中，使用val接口可输出模型的层数、参数量、计算量。而与推理速度相关的评价指标中，只提供了前处理、推理和后处理的耗时，并且是一次推理的的耗时，而单次推理的耗时仍可能因各种随机因素波动，导致单次测量结果不可靠。本文通过预热、取平均，可一键输出模型的推理延时、FPS以及层数、参数量、计算量、模型大小。

---

专栏目录：RT-DETR改进目录一览 | 涉及卷积层、轻量化、注意力、损失函数、Backbone、SPPF、Neck、检测头等全方位改进

专栏地址：RT-DETR改进专栏——以发表论文的角度，快速准确的找到有效涨点的创新点！

一、完整代码

在项目的根目录中新建一个py文件，将下面的代码粘贴进去即可。

相关的参数参考第二节进行调整。

import warnings

warnings.filterwarnings('ignore')
import argparse
import os
import time
import numpy as np
import torch
from tqdm import tqdm
from ult