The code base you are using assumes that you are providing an image mask, not coordinates.
If you don’t have this type of mask, I would recommend to write a custom Dataset, loading the input images and the corresponding head positions.