GroundingDINO onnxrun

Ao executar o programa, preste atenção ao formato das palavras solicitadas por entrada, separe as categorias com "." E verifique se o nome da categoria existe na categoria Dictionary File Vocab.txt, e o nome da categoria na palavra prompt de entrada é a categoria de destino que você deseja detectar, caso contrário, o alvo não pode ser detectado.

Se você deseja exportar o arquivo onnx, execute export_onnx.py em https://github.com/wenyi5608/grounddino para gerar o arquivo onnx. Observe que a versão Pytorch precisa estar acima de 2.0. A diferença entre o código neste repositório e o código no repositório oficial https://github.com/idea-research/groundingdino é que os parâmetros de entrada da função avançada nos modelos Grounddino Models Groundingdino Groundingdino.py são diferentes.

O arquivo ONNX exportado está no Baidu Cloud Drive, Download Link: https://pan.baidu.com/s/1_dxasmg2vbw47fj7fduug Código de extração: U6LR