SDT icon indicating copy to clipboard operation
SDT copied to clipboard

Style_score的问题

Open SupremeU opened this issue 9 months ago • 4 comments

SupremeU avatar Mar 29 '25 08:03 SupremeU

Image 生成128x128的图片测Style_score 为什么是这个结果 sk_pil = coords_render(preds[i], split=True, width=128, height=128, thickness=8, board=0)

SupremeU avatar Apr 01 '25 06:04 SupremeU

因为风格分数模型不是在128*128的数据上训练的,这样评测会有很大的泛化误差。

dailenson avatar Apr 24 '25 02:04 dailenson

但是用64x64会tensor形状不匹配

SupremeU avatar Apr 24 '25 02:04 SupremeU

sk_pil = coords_render(preds[i], split=True, width=48, height=48, thickness=1, board=0)
sk_pil = sk_pil.resize((113, 113))

用上述代码替换下面这行代码即可: https://github.com/dailenson/SDT/blob/339aa2155d93263bdb7d533830559d011d41995c/test.py#L94

dailenson avatar Apr 24 '25 03:04 dailenson