GroundingDINO onnxrun
1.0.0
在運行程序時,要注意輸入的提示詞的格式,類別之間以" . "隔開,並且確保類別名稱在詞典文件vocab.txt裡是存在的,而且輸入提示詞裡的類別名稱是你想要檢測的目標類別,否則可能會檢測不到目標的。
如果要導出onnx文件,把export_onnx.py放在https://github.com/wenyi5608/GroundingDINO 裡運行就可以生成onnx文件的,注意pytorch的版本需要在2.0以上的。這個倉庫裡的代碼跟官方倉庫https://github.com/IDEA-Research/GroundingDINO 裡的代碼的不同之處在於groundingdinomodelsGroundingDINOgroundingdino.py裡的forward函數的輸入參數不同。
已經導出的onnx文件在百度雲盤,下載鏈接:https://pan.baidu.com/s/1_dDxaSMG2vbw47FJ7FdUUg 提取碼:u6lr