Add classes/ignore_index to losses

isaaccorley / torchseg

Segmentation models with pretrained backbones. PyTorch.

MIT License

104 stars 8 forks source link

Add classes/ignore_index to losses #11

Closed isaaccorley closed 8 months ago

isaaccorley commented 8 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