elasticdeform.tf

TensorFlow wrapper function

elasticdeform.tf.deform_grid(X, displacement, *args, **kwargs)[source]

Elastic deformation with a deformation grid, wrapped in a TensorFlow Op.

This function wraps the elasticdeform.deform_grid function in a TensorFlow Op with a custom gradient.

Parameters:

X (Tensor or list of Tensors) – input image or list of input images
displacement (Tensor or numpy array) – displacement vectors for each control point

Returns:

the deformed image, or a list of deformed images

Return type:

Tensor