tensorflow - 不平衡数据集对象检测

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
}