Skip to content

使用onnxruntime部署GroundingDINO开放世界目标检测,包含C++和Python两个版本的程序

Notifications You must be signed in to change notification settings

decision-labs/GroundingDINO-onnxrun

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

11 Commits
 
 
 
 
 
 
 
 

Repository files navigation

在运行程序时,要注意输入的提示词的格式,类别之间以" . "隔开,并且确保类别名称在词典文件 vocab.txt里是存在的,而且输入提示词里的类别名称是你想要检测的目标类别,否则可能会检测不到目标的。

如果要导出onnx文件,把export_onnx.py放在https://github.com/wenyi5608/GroundingDINO 里运行就可以生成onnx文件的,注意pytorch的版本需要在2.0以上的。这个仓库里的代码跟官方仓库https://github.com/IDEA-Research/GroundingDINO 里的代码的不同之处在于 groundingdino\models\GroundingDINO\groundingdino.py里的forward函数的输入参数不同。

已经导出的onnx文件在百度云盘,下载链接:https://pan.baidu.com/s/1_dDxaSMG2vbw47FJ7FdUUg 提取码:u6lr

About

使用onnxruntime部署GroundingDINO开放世界目标检测,包含C++和Python两个版本的程序

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 64.9%
  • C++ 35.1%