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

Question

我想将我的车辆图像(90、180 或 270 度)旋转到正确的位置。
我使用了一个假设:将图像旋转 4 个角度 (0, 90, 180, 270) 并将每个角度赋予一个对象检测神经网络，给出最大汽车对象置信度得分的角度将是我必须的角度用于旋转我的图像。
问题是:一些 180 度旋转的图像比原始图像给出更好的分数，因为 NN 错误地检测到有 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

在图片中搜索车轮并确定它们的中心是在车箱上方还是下方。