GroundingDINO-onnxrun
GroundingDINO-onnxrun copied to clipboard
使用onnxruntime部署GroundingDINO开放世界目标检测,包含C++和Python两个版本的程序
你好,我跑你的实现,CPU版本没有问题,但是设置为GPU模式时报下面的错误: 2024-03-19 08:12:34.290537612 [E:onnxruntime:, sequential_executor.cc:514 ExecuteKernel] Non-zero status code returned while running Expand node. Name:'/class_embed.0_4/Expand' Status Message: /class_embed.0_4/Expand: left operand cannot broadcast on dim 2 LeftShape: {1,900,8}, RightShape: {1,900,256} terminate...
结果与原版不对齐
原版跑出来结果更准确
c++ 版本推理结果 score 都被阈值过滤掉, score都很小