GroundingDINO onnxrun

Lorsque vous exécutez le programme, faites attention au format des mots d'invite d'entrée, séparez les catégories avec ".", Et assurez-vous que le nom de catégorie existe dans le fichier de dictionnaire Vocab.txt, et le nom de catégorie dans le mot d'invite d'entrée est la catégorie cible que vous souhaitez détecter, sinon la cible peut ne pas être détectée.

Si vous souhaitez exporter le fichier ONNX, exécutez Export_onnx.py dans https://github.com/wenyi5608/groundingDino pour générer le fichier ONNX. Notez que la version Pytorch doit être supérieure à 2.0. La différence entre le code de ce référentiel et le code dans le référentiel officiel https://github.com/idea-research/groundingDino est que les paramètres d'entrée de la fonction avant dans Grounddino Models FarmingDino FarmingDino.py sont différents.

Le fichier ONNX exporté est dans Baidu Cloud Drive, Télécharger le lien: https://pan.baidu.com/s/1_ddxasmg2vbw47fj7fduug Code d'extraction: U6LR