MoGe icon indicating copy to clipboard operation
MoGe copied to clipboard

为什么我自己本地跑的结果和你们网站HF demo跑出来的不一致呢

Open RanYu-GN opened this issue 6 months ago • 4 comments

第一个是我在你们网站中跑出来的结果,看起来很好

Image

下面这个是我用git clone下来你们的源码,本地3090运行的结果,我们用的是 moge/scripts/infer.py 跑的,模型我们使用了你们下面这两个模型

Image

但是结果是这样的,这看起来很不好,这个原因是什么呢?

Image

RanYu-GN avatar Jun 13 '25 03:06 RanYu-GN

我也出现了这个问题

chendalao avatar Jun 13 '25 04:06 chendalao

网站是MoGe-1的结果。本地现在默认跑的MoGe-2由于训练数据的不同,mask prediction的偏好会和MoGe-1不一样。这个pattern是mask估计错误导致的(无法判断背景应该作为实体墙保留,还是应该作为非实体的背景去掉)。如果发现mask挂掉,可以将apply_mask置为False,只取原始depth。

EasternJournalist avatar Jun 13 '25 06:06 EasternJournalist

好的,谢谢,我将apply_mask 改为了 False,这样确实是depth看起来很好了,但是我想知道,我这样修改之后,会不会导致有些结果变差呢?

RanYu-GN avatar Jun 13 '25 07:06 RanYu-GN

精度不会变差的,apply_mask的区别只有是否把天空或纯色背景的深度替换为inf,不影响深度预测的结果。不过对于确实存在天空的图片,如果不apply_mask,天空区域会保留无意义的深度,看起来会比较奇怪。

EasternJournalist avatar Jun 25 '25 06:06 EasternJournalist