不平衡数据集目标检测

Question

对于具有高度倾斜数据的数据集，有没有好的方法来微调用于对象检测的模型(特别是，我正在尝试使用Tensorflow对象检测API)？我正在尝试使用COCO的一些类别，并将其与我自己的自定义数据相结合，但我的数据只有大约50张图像。

我尝试过将coco数据和我自己的数据结合起来，但每次它都只能预测coco类别。

Answer 1

你可以尝试使用焦点损失。

请参阅：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
}