mmdetection框架 實現cross_entropy多分類類加權

注意:我使用的mmdetection並非最新版本。 位置:mmdet/models/losses/cross_entropy_loss.py 代碼如下:增加了weight部分 label是element-wise的,所以weight也做成element-wise即可,後面進入weight_reduce_loss函數中會loss和weight相乘。
相關文章
相關標籤/搜索