search

hustvl / ViTMatte

[Information Fusion (Vol.103, Mar. '24)] Boosting Image Matting with Pretrained Plain Vision Transformers
MIT License
363 stars 37 forks source link

你好，我想向你请教一下，关于测量指标的问题。 #33

Closed davislee546 closed 6 months ago

davislee546 commented 6 months ago

你好，我这两天跑评估的时候看到指标只有SAD和MSE两个。论文里面我看到有很多的指标对比，包括Conn和Grad这些。还有FLOPs和param这些对比。请问你是怎么增加这些指标的呀？我自己加包结果导致环境坏了。。。

JingfengYao commented 6 months ago
  1. 最终论文的指标使用matlab测的，主要原因是使用python测试的结果会和DIM论文的结果略有差别。
  2. 一般实验情况下，通常只会看SAD和MSE的指标（如MGMatting, MatteFormer），后两个指标和它们基本是正相关的。如果你想用python测grad和conn，你可以参考之前的一些方法：GCAMatting。
  3. 由于vitmatte的架构是d2搭建的，可以相对方便的调用d2的测试代码analyze_model来测试模型参数量，flops等等信息