SDT
SDT copied to clipboard
Style_score的问题
生成128x128的图片测Style_score 为什么是这个结果
sk_pil = coords_render(preds[i], split=True, width=128, height=128, thickness=8, board=0)
因为风格分数模型不是在128*128的数据上训练的,这样评测会有很大的泛化误差。
但是用64x64会tensor形状不匹配
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