Add classes/ignore_index to losses

isaaccorley / torchseg

Segmentation models with pretrained backbones. PyTorch.

MIT License

91 stars 7 forks source link

Add classes/ignore_index to losses #11

Closed isaaccorley closed 4 months ago

isaaccorley commented 4 months ago

Some of the loss functions use either ignore_index, classes, or in some cases both. However some losses are are able to support both. This PR adds the following:

Support classes in FocalLoss
Support ignore_index in JaccardLoss

See https://github.com/isaaccorley/torchseg/issues/7