search

eriklindernoren / PyTorch-YOLOv3

Minimal PyTorch implementation of YOLOv3
GNU General Public License v3.0
7.33k stars 2.63k forks source link

how to change the loss function of this model? #367

Open gjt9274 opened 4 years ago

gjt9274 commented 4 years ago

there is a sample imbalance problem in my data(with positive and negative ratio is 1:10)，so I wanna to change the loss function to focal loss. anybody who knows where to change the code and how or another solution to solve my question? Thanks a lot

songxiaoju commented 4 years ago

I want to do this, too.

gjt9274 commented 4 years ago

I want to do this, too.

so, did you solve it?

songxiaoju commented 4 years ago

I want to do this, too.

so, did you solve it?

sorry,I can't deal with it.