Add support for object categories

jpsember / java-ml

Java classes for machine learning

0 stars 0 forks source link

Add support for object categories #59

Closed jpsember closed 2 years ago

jpsember commented 2 years ago

I haven't incorporated this into the yolo loss function yet.

jpsember commented 2 years ago

Relevant: https://pytorch.org/docs/stable/generated/torch.nn.CrossEntropyLoss.html

jpsember commented 2 years ago

[x] Is there some sort of normalization I should apply to the crossentropy loss?

jpsember commented 2 years ago

https://stackoverflow.com/questions/49390842/cross-entropy-in-pytorch

PyTorch's CrossEntropyLoss expects unbounded scores (interpretable as logits / log-odds) as input, not probabilities (as the CE is traditionally defined).