FocalLoss-MXNet

KO | EN

gitlite — search

#ai #llm #rust #deep-learning #docker #macos #java #cli #kubernetes #awesome-list #security #ios #mcp #openai

FocalLoss-MXNet

(★ 50)

No description available.

FocalLoss-MXNet Latest Version Download

Download Latest Version (.zip)

// repository documentation

Was this content helpful?

★ 0 (0 ratings)

### This is the optimized version of focal loss in MXNet, which is modified from [unsky/focal-loss](https://github.com/unsky/focal-loss) and speed up 30% than Original implement during the training.

* The use of focal loss is same as [unsky/focal-loss](https://github.com/unsky/focal-loss))

```
from focal_loss_OptimizedVersion import *
label = mx.sym.Variable('focalloss_label')
net = mx.symbol.Custom(data=net, op_type='FocalLoss', labels = label, name='focalloss', alpha=0.25, gamma=2)
```

* Apart from `focal_loss_OptimizedVersion.py`, I alse provide `metric.py` for presenting focal loss value by taking image classification as example:

```
from metric import *
eval_metric = mx.metric.CompositeEvalMetric()
eval_metric.add(FocalLoss())

model = mx.mod.Module(
        context=mx.gpu(0),
        symbol=symbol,
        label_names=('focalloss_label',)
    )

model.fit(...,
	  eval_metric=eval_metric,
	  ...)
```

Attention: The value of alpha and gamma in `metric.py` should be equal to `mx.symbol.Custom(...,alpha, gamma)`

FocalLoss-MXNet

FocalLoss-MXNet Latest Version Download

Today's picks

Was this content helpful?

Recent Feedback

FocalLoss-MXNet

FocalLoss-MXNet Latest Version Download

Today's picks

Was this content helpful?

Recent Feedback

Download README