mmdetection框架 实现cross_entropy多分类类加权

注意:我使用的mmdetection并非最新版本。

位置:mmdet/models/losses/cross_entropy_loss.py

代码如下:增加了weight部分
在这里插入图片描述
label是element-wise的,所以weight也做成element-wise即可,后面进入weight_reduce_loss函数中会loss和weight相乘。

展开阅读全文

没有更多推荐了,返回首页

©️2019 CSDN 皮肤主题: 大白 设计师: CSDN官方博客
应支付0元
点击重新获取
扫码支付

支付成功即可阅读