Skip to content
New issue

Have a question about this project? # for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “#”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? # to your account

Fix ONNX demo postprocessing #513

Open
wants to merge 1 commit into
base: master
Choose a base branch
from

Conversation

artur-ag
Copy link

Same fix as here from @shiboyang

Tested with

python deploy/export_onnx.py \
  configs/pretrain/yolo_world_v2_l_vlpan_bn_2e-3_100e_4x8gpus_obj365v1_goldg_train_lvis_minival.py \
  demo/pretrained_weights/yolo_world_v2_l_obj365v1_goldg_pretrain-a82b1fe3.pth \
  --custom-text custom-text.json \
  --opset 12 --without-nms

python deploy/onnx_demo.py \
  work_dirs/yolo_world_v2_l_obj365v1_goldg_pretrain-a82b1fe3.onnx \
  <INPUT> custom-text.json \
  --onnx-nms

# for free to join this conversation on GitHub. Already have an account? # to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant