elasticdeform.tf

TensorFlow wrapper function

elasticdeform.tf.deform_grid(X, displacement, *args, **kwargs)

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

See also

elasticdeform.deform_grid()

for the other parameters