GroundingDINO onnxrun

Al ejecutar el programa, preste atención al formato de las palabras indicadoras de entrada, separe las categorías con ".", Y asegúrese de que el nombre de la categoría exista en el archivo de diccionario Vocab.txt, y el nombre de la categoría en la palabra del mensaje de entrada es la categoría de destino que desea detectar, de lo contrario el objetivo no puede ser detectado.

Si desea exportar el archivo ONNX, ejecute export_onnx.py en https://github.com/wenyi5608/groundingdino para generar el archivo ONNX. Tenga en cuenta que la versión de Pytorch debe estar por encima de 2.0. La diferencia entre el código en este repositorio y el código en el repositorio oficial https://github.com/idea-research/groundingdino es que los parámetros de entrada de la función de reenvío en Grounddino Models Grounddino Groundingdino.py son diferentes.

El archivo ONNX exportado se encuentra en Baidu Cloud Drive, Descargar enlace: https://pan.baidu.com/s/1_ddxasmg2vbw47fj7fduug Código de extracción: U6LR