使用thop库分析模型的参数量、运算量、帧率

Parameters    参数量，指模型含有多少参数。
GFLOPs    FLOPs 是浮点运算次数，可以用来衡量算法/模型复杂度GFLOPs。为十亿（1e9）次的浮点运算。
Latency    网络前向传播的时间，1 ms=1e-3 s，10.5ms=0.0105s
FPS    每秒传输帧数，FPS=1/Latency，1/0.0105=95.2

首先安装thop

pip install thop

测试模型参数量和Flops

    from thop import profile
    network = SwinTransformer(
    hidden_dim=96,
    layers=(2, 2, 18, 2),
    heads=(3, 6, 12, 24),
    channels=3,
    num_classes=8,
    head_dim=32,
    window_size=16,
    downscaling_factors=(4, 2, 2, 2),
    relative_pos_embedding=True
    )
    input = torch.randn(3,3,512,1024)
    
    flops, params = profile(network, inputs=(input, ))
    print("flops",flops)
    print("params",params)

测试一张图所预测的时间：

    network.cuda()
    network.eval()
    imgs = input.cuda()
    start = time.time()
    outputs,*outputs_aux = network(imgs)
    end = time.time()
    print("latency: ", end-start)
    print("fps: ", 1/(end-start))