Multi-dimensional Labels in pytorch
sarthak1996
apaszke
I have some co-ordinates for bounding boxes in the train dataset and I want the net to train on those.
sarthak1996
What kind of loss function do you want to use then?
apaszke
sarthak1996
Sooo you’re predicting a 0 or 1 for each pixel? Why does your output have 3 channels then?
apaszke
L1 and L2 losses should be easy to write yourself using only autograd operations. It’s just a subtraction + abs or pow.
apaszke