不平衡的数据集对象检测

Question

是否存在一种很好的方法来微调对象检测模型（特别是我正在尝试使用Tensorflow对象检测API）用于数据高度偏斜的数据集？我正在尝试使用某些类别的COCO，并将其与我自己的自定义数据结合起来，但是我的数据只有大约50张图像。

我尝试仅将coco数据和我自己的数据组合在一起，但是它每次都能预测coco类别。

Answer 1

您可以尝试使用Focal Loss。

请参阅：https://arxiv.org/pdf/1708.02002.pdf

在Tensorflow对象检测模型文件中，其显示如下：

loss {
    localization_loss {
        weighted_smooth_l1 {
        }
    }
    classification_loss {
        weighted_sigmoid_focal {
            gamma: 2.0
            alpha: 0.25
        }
    }
    classification_weight: 1.0
    localization_weight: 1.0
}