Yes, I encountered a similar issues with the first iteration of dataset. The prediction location of the needle tip would be wildly off when the view angle was high. The fix was easy, just generated high viewing angle images for the 2nd dataset.