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

注意:我使用的mmdetection并非最新版本。 位置:mmdet/models/losses/cross_entropy_loss.py 代码如下:增加了weight部分 label是element-wise的,所以weight也做成element-wise即可,后面进入weight_reduce_loss函数中会loss和weight相乘。
相关文章
相关标签/搜索