#964·CenterNet

Ignore objects in loss calculation

Author: r3krutCreated Jan 16, 2022Updated Jul 5, 2024

Hi! Thanks for your outstanding work! I have a some question about loss calculation. Suppose, I have in my dataset some target objects which should be ignored during training. There is a right way to do this during heatmap loss calculation based on Gaussians?

Should I generate a binary training mask for pixels which belongs to "ignore" object for whole gaussian(positive+negative) or only for center(positive) pixel?

This is original image with targets visualization(red - targets, cyan - ignoring objects. oimg

This is source heatmaps for all objects. ohm

This mask for ignoring objects(drawing as circle with same radius as in heatmap generation) m

This is masked source heatmaps. mhm

My question is: is there a proper way to generate training mask for ignoring objects?