search

WongKinYiu / yolov7

Implementation of paper - YOLOv7: Trainable bag-of-freebies sets new state-of-the-art for real-time object detectors
GNU General Public License v3.0
13.41k stars 4.23k forks source link

Change class weights #2060

Open Artem-N opened 4 months ago

Artem-N commented 4 months ago

Hi, is it possible to affect on class weights due to the imbalance of my dataset ( 3 classes have small size and 3 balanced), I would change them accordingly so that the model learns better, or does it automatically in the loss function in YOLOv7?

eraydikyologlu commented 4 months ago

what did you do? Did you have a solution for this?

Artem-N commented 2 months ago

what did you do? Did you have a solution for this?

unfortunately not, I supplemented the classes with images and hope that Yolo7 is smart and will be able to balance the class weights itself. Because it seems to me that Yolo can adjust the weights of the classes itself.