A method includes, in a processor, receiving example representations of geometrical shapes of a given type of organ. In a training phase, a neural network model is trained using the example representations. In a modeling phase, the trained neural network model is applied to a set of location measurements acquired in an organ of the given type, to produce a three-dimensional model of the organ.
