Skip to content

Latest commit

 

History

History
12 lines (10 loc) · 716 Bytes

README.md

File metadata and controls

12 lines (10 loc) · 716 Bytes

Focal loss

Usage

  • firstly, you should get a list which contains each class number, like classes_nu=[1,2,3] means index_0 class have 1 pic, index_1 class have 1 pics, index_2 class have 3 pics.
  • then, use the focal loss function like below:

model.compile(optimizer=SGD(lr=learning_rate, momentum=0.9), loss=[focal_loss(classes_num)], metrics=['accuracy'])

blog of focal loss