AlibabaResearch / AlibabaResearch/AdvancedLiterateMachinery

VGT模型配置中,VOCAB_SIZE也许应该为30522而不是30552

未关闭
#83 0 条评论 1 个 reaction 已指派 0 人 在 GitHub 查看
主要语言
C++
星标
1.8k
派生
195
PR 合并指标
30 天内没有已合并 PR

描述

下载Readme中D4LA对应模型,并执行inference.py

`
python inference.py \
--image_root 'xxx' \
--grid_root 'xxx' \
--image_name 'budget_0000022278' \
--dataset D4LA \
--output_root output/ \
--config Configs/cascade/D4LA_VGT_cascade_PTM.yaml \
--opts MODEL.WEIGHTS model/D4LA_VGT_model.pth \
MODEL.WORDGRID.USE_PRETRAIN_WEIGHT False
`

产生一条警告,且输出的图片中没有任何检测结果:

`
Skip loading parameter 'Wordgrid_embedding.embedding.weight' to the model due to incompatible shapes: (30522, 768) in the checkpoint but (30552, 768) in the model! You might want to double check if this is expected.
`

修改yaml文件中的VOCAB_SIZE为30522后,不再产生警告,并且输出图片有了检测结果框。

贡献指南

这个仓库没有索引到贡献指南

评估

这个 Issue 还没有评估数据。

把新 issue 发到你的邮箱

精选适合新手参与的 GitHub issue 摘要。