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

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

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

代码如下:增加了weight部分
mmdetection框架 实现cross_entropy多分类类加权
label是element-wise的,所以weight也做成element-wise即可,后面进入weight_reduce_loss函数中会loss和weight相乘。