image-processing - 检测旋转的车辆图像

Question

我想将车辆图像(90、180 或 270 度)旋转到正确的位置。
我使用的假设是:将图像旋转 4 个角度(0、90、180、270)并将每个角度提供给对象检测神经网络，给出最大汽车对象置信度分数的角度将是我必须的角度用于旋转我的图像。
问题是:一些 180 度旋转的图像比原始图像给出更好的分数，因为神经网络错误地检测到有 2 个轮子的道路是汽车。例如:
The original image
The rotated image but get better score

我怎样才能防止这种情况发生？
我的模型:ssd_resnet_50_fpn_coco 来自 model zoo
以下是其他一些失败案例:
1 2 enter image description here

Answer 1

在图片中搜索车轮，并确定其中心是在车箱上方还是下方。