Text this: Neural network based 3D tracking with a graphene transparent focal stack imaging system