@@ -12,7 +12,7 @@ A fast tool to do image augmentation by CUDA on GPU(especially elastic deform),
12
12
- Very Fast, speed up 25x at rotation, 45x at elastic_deform.
13
13
- Support many types of spatial deform: flip, rotate, scale, translate, elastic_deform.
14
14
- Support many rules of map_coordinates: mirror, constant, reflect, wrap, nearest.
15
- - Doing Spation_Deform by doing calculates on coordinates, all transformations get combined before they are applied to the image
15
+ - Doing Spatial_Deform by doing calculates on coordinates, all transformations get combined before they are applied to the image
16
16
- Implement map_coordinates by linear interpolation.
17
17
- Unit test pass when over 99% pixels has L1_loss < 1e-3.
18
18
- Flexible with coordinates, users can fetch them form CUDA and do cubic interpolation at CPU by scipy.map_coordinates(order = 3)
@@ -44,7 +44,7 @@ cuda_handle = Cuda_Spatial_Deform(array_image.shape, mode="constant")
44
44
'''
45
45
Shape: cuda_backend will malloc according to shape
46
46
RGB: bool (Only Support 2D-RGB)
47
- mode: The rules of map_coordinates. Refernce to https://docs.scipy.org/doc/scipy/reference/generated/scipy.ndimage.map_coordinates.html
47
+ mode: The rules of map_coordinates. Reference to https://docs.scipy.org/doc/scipy/reference/generated/scipy.ndimage.map_coordinates.html
48
48
cval: default is 0.0. Only be useful when mode == 'constant'
49
49
id_gpu: choose the number of GPU
50
50
'''
0 commit comments