With the data augmentation, it’s very likely you’re losing points because they fall off the picture. You either need to:
- use less data augmentaiton
- use the option
remove_out=FalseinImagePoints(to not remove them), beware that you will have targets outside of -1,1, so adapt the scaling of your last layer accordingly. - pad your inputs to deal with those missing points