Skip to content

Add support to only keep selected classes in one_hot #3581

@Nic-Ma

Description

@Nic-Ma

Is your feature request related to a problem? Please describe.
Feature request from @holgerroth:
I'm looking for something similar to the question asked here https://discuss.pytorch.org/t/make-one-hot-encoding-with-ignore-label-for-semantic-segmentation/18126/10
The background can be thought of as a ignored label. We could even have a list of labels to ignore when converting to one-hot, I think.
Predicting the background in the network is basically redundant in this case.

Metadata

Metadata

Assignees

Labels

enhancementNew feature or request

Type

No type

Projects

Relationships

None yet

Development

No branches or pull requests

Issue actions