AlibabaResearch / AlibabaResearch/AdvancedLiterateMachinery

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

オープン
#83 コメント 0 件 リアクション 1 件 担当者 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 を短くまとめたダイジェスト。