WongKinYiu / yolor

implementation of paper - You Only Learn One Representation: Unified Network for Multiple Tasks (https://arxiv.org/abs/2105.04206)
GNU General Public License v3.0
1.99k stars 518 forks source link

Class weights supported? #231

Open Swearinbag opened 2 years ago

Swearinbag commented 2 years ago

I was wondering, since YOLOV3-CSP is the backbone of the explicit model, is configuring class weights a supported feature in YOLOR?