Related to Focal Loss Layer: is it suitable for multi-class classification?

1 次查看(过去 30 天)
Does the focal loss layer (in Computer vision toolbox) support multi-class classification (or suited for binary prolems only)?

回答(1 个)

Hrishikesh Borate
Hrishikesh Borate 2021-4-20
Hi,
The focalLossLayer from the Computer Vision Toolbox does support multi-class classification problems. The classes can be defined during the creation of focalLossLayer using ‘Classes’ property, as shown below.
classes = ["class1", "class2", "class3", "class4"];
layer = focalLossLayer('Classes',classes,'Name','focallosslayer');
For more information, refer to focalLossLayer.

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!

Translated by