张量流对象检测中类失衡的加权损失更快rcnn

pra*_*tsJ 6 object-detection tensorflow

我正在尝试使用更快的RCNN架构的tensorflow对象检测代码,并且由于数据集中的高级别不平衡,我试图将损失修改为加权损失(基于不同类图像的分数).但是代码中已经存在softmax损失的权重,最有可能处理正例和背景之间的类不平衡(负).如何修改此损失以解决班级间的不平衡问题,而不仅仅是课堂与背景(非班级)之间?