我的模型有2个班级(没有背景班级),并通过ssd_mobilenet_v2_coco的迁移学习进行了训练。它可以很好地检测和分类训练对象。但是,在新图像上,它还会检测到背景的许多假阳性边界框。 有时,检测到的背景对象是可可数据集中存在的对象(例如杯子),但不仅是-黑屏也被检测为对象。
这是我正在使用的配置:
hard_example_miner {
num_hard_examples: 3000
iou_threshold: 0.99
loss_type: CLASSIFICATION
max_negatives_per_positive: 3
in_negatives_per_image: 3
}
我应该对此做任何更改吗?还是有其他解决方案可以改善背景图片的模型?