我写了一个yolov5_deepsort的tensorrt版本，测试的时候发现deepsort无效，yolov5可以正常预测；请问可以交流交流吗？

[Wulingtian]

[Wulingtian]

已解决

[huhuhu12138]

可不可以 跟您这边请教一下，，

[kundebuxing]

已解决

请问怎么结合yolov5 tensorrt 和deepsort？

[cuiyong127]

请问怎么实现的呢？可以分享下吗？

[huhuhu12138]

加个微信说吧，13835171347

---原始邮件--- 发件人: @.> 发送时间: 2021年8月2日(周一) 下午2:18 收件人: @.>; 抄送: @.@.>; 主题: Re: [shaoshengsong/DeepSORT] 我写了一个yolov5_deepsort的tensorrt版本，测试的时候发现deepsort无效，yolov5可以正常预测；请问可以交流交流吗？ (#14)

请问怎么实现的呢？可以分享下吗？

— You are receiving this because you commented. Reply to this email directly, view it on GitHub, or unsubscribe.

[FishHe]

我在copy 代码的时候发现数据转换的代码有问题，不知道是不是相关，放在这里： void tobuffer(const std::vectorcv::Mat& imgs, wa::Image* image) { int pos = 0; for (const cv::Mat& img : imgs) { int Lenth = img.rows * img.cols * 3; int nr = img.rows; int nc = img.cols; if (img.isContinuous()) { nr = 1; nc = Lenth; } for (int i = 0; i < nr; i++) { const uchar* inData = img.ptr(i); for (int j = 0; j < nc; j++) { buf[pos] = *inData++; pos++; } }//end for }//end imgs; }

最内层的数据复制代码有误，应改为： int pos = 0; for (const cv::Mat& img : imgs) { int Lenth = img.rows * img.cols; int nr = img.rows; int nc = img.cols; if (img.isContinuous()) { nr = 1; nc = Lenth; } for (int i = 0; i < nr; i++) { const uchar* inData = img.ptr(i); for (int j = 0; j < nc; j++) { buf[pos++] = *inData++; buf[pos++] = *inData++; buf[pos++] = *inData++; } }//end for }//end imgs;

[huangqi111]

请问你下，我用自己的数据集，怎么训练feature.onnx,是用什么代码，能提供下么