search

tztztztztz / eqlv2

The official implementation of Equalization Loss v1 & v2 (CVPR 2020, 2021) based on MMDetection. https://arxiv.org/abs/2012.08548 https://arxiv.org/abs/2003.05176
Apache License 2.0
155 stars 22 forks source link

Implementation of category-agnostic task #14

Open IssacCyj opened 2 years ago

IssacCyj commented 2 years ago

Thanks for this great repo.

It mentioned in the paper that you also used category-agnostic task. Is it implemented in this repo? I didn't find it but I was able to reproduce the results in the paper.

tztztztztz commented 2 years ago

The output shape of the classifier is 1231: 1230 (category-specific task) + 1 (category-agnostic task).

z3n0e commented 2 years ago

Does eqlv2 work without the category-agnostic task? I'm trying to implement it on yolov5...

tztztztztz commented 2 years ago

Yes. BTW, if you want to use focal loss, you can use equalized focal loss