search

Zhou-sx / yolov5_Deepsort_rknn

Track vehicles and persons on rk3588 / rk3399pro.
340 stars 83 forks source link

得到fastreid模型的特征向量 #9

Open Takizzl opened 2 years ago

Takizzl commented 2 years ago

博主您好，我想问一下您的code里面有图片送入模型后得到的相关特征向量的代码吗，不知道为什么您在getRectsFeature函数中返回的是bool型呢，没有返回feature这个值吗，因为我想要得到feature特征向量，来和库中的特征向量计算其余弦相似度，得到我的结果

Zhou-sx commented 2 years ago

奥，getRectsFeature函数不应该有返回值，引起了歧义 getRectsFeature函数里 FeatureTensor::doInference计算了feature特征向量 放在了det[i].feature里面(det[i] 是一个DETECTION_ROW类)