search

Aimol-l / OrtInference

Using OnnxRuntime to inference yolov10,yolov10+SAM ,yolov10+bytetrack and SAM2 by c++ .
MIT License
58 stars 7 forks source link

推理失败想请教一下可能的问题 #9

Closed Anivie closed 2 weeks ago

Anivie commented 2 weeks ago

您好，想问问您是否遇到过这样的情况：执行推理后输出的图片异常 a_out 我理解的单张图片推理应该是这个流程：

图片拉伸缩放->通道修改为RGB->像素归一化->输入到image_encoder->点+点标签+图片原尺寸+image_encoder[vision_feats]+image_encoder[high_res_feat0]+image_encoder[high_res_feat1]输入到image_decoder->pred_mask的输出就是mask，乘以像素就能直接看到mask的轮廓

我理解的这个流程有误吗？希望能指导一下

Aimol-l commented 2 weeks ago

我的单张图片推理流程：cv读取一张图片 --> 直接resize为1024*1024 or 使用padding策略 -->直接除255 or 减mean除std --> 调整维度到 [1,3,1024,1024] -->image_encoder --> image_decoder -->取出 pred_mask --> 后续处理...

Anivie commented 2 weeks ago

抱歉，思路应该是对的，不知道哪一步出了问题一直出不来，换了几个函数就成功了