我正在使用的模型(LayoutLM[seq_labeling]):在使用FUNSD数据集成功训练后，当我尝试进行预测时，我只得到了test_predictions,而不是边界框坐标。请问您能帮我在预测时得到边界框吗？
n3schb8v1#
它的输出顺序与传入的输入标记相同。您可以轻松地将边界框Map到相应的预测标记，因为它们由换行符分隔。
mgdq6dx12#
@SibtainRazaJamali @sryeddula 你能分享一下运行 --do_predict 命令时使用的命令吗？另外，我们是否需要在 FUNSD 上重新训练，还是可以使用预训练的权重进行预测？
2条答案
按热度按时间n3schb8v1#
它的输出顺序与传入的输入标记相同。您可以轻松地将边界框Map到相应的预测标记，因为它们由换行符分隔。
mgdq6dx12#
@SibtainRazaJamali @sryeddula 你能分享一下运行 --do_predict 命令时使用的命令吗？另外，我们是否需要在 FUNSD 上重新训练，还是可以使用预训练的权重进行预测？