You will have remap the labels and update the txt files so that the class ID for person
in your dataset is 0 because that’s what the pretrained models use for persons
.
Right now, it seems head
is your first class so it would have class ID 0 and person is second class, so it would have class ID 1.
After remapping, you can pass classes=0
like you did to measure performance on person
class.