search

pytorch / android-demo-app

PyTorch android examples of usage in applications
1.47k stars 606 forks source link

目标检测精度损失问题反馈 #327

Closed RoAdv closed 1 year ago

RoAdv commented 1 year ago

您好，我们在运行ObjectDetection的demo时，检测的效果与PC上的效果不一致，然后阅读代码，发现有两个地方存在bug，想跟您探讨下：

  1. PrePostProcessor#nonMaxSuppression的排序算法中，描述的是从高到低降序，但实际是升序，导致会优先筛选出置信度低的结果
  2. ObjectDetectionActivity#analyzeImage中有一行 matrix.postRotate(90.0f)为什么要旋转90度呢？如果原图训练的时候没有做旋转90度处理，但是检测的时候做该处理就会导致检测有误，置信度极低