zoo.feature.image3d package

Submodules

zoo.feature.image3d.transformation module

class zoo.feature.image3d.transformation.AffineTransform3D(affine_mat, translation=array([0., 0., 0.]), clamp_mode='clamp', pad_val=0.0, bigdl_type='float')

Bases: zoo.feature.image3d.transformation.ImagePreprocessing3D

Affine transformer implements affine transformation on a given tensor. To avoid defects in resampling, the mapping is from destination to source. dst(z,y,x) = src(f(z),f(y),f(x)) where f: dst -> src :param affine_mat: numpy array in 3x3 shape.Define affine transformation from dst to src. :param translation: numpy array in 3 dimension.Default value is np.zero(3). Define translation in each axis. :param clampMode: str, default value is “clamp”. Define how to handle interpolation off the input image. :param padVal: float, default is 0.0. Define padding value when clampMode=”padding”. Setting this value when clampMode=”clamp” will cause an error.

class zoo.feature.image3d.transformation.CenterCrop3D(crop_depth, crop_height, crop_width, bigdl_type='float')

Bases: zoo.feature.image3d.transformation.ImagePreprocessing3D

Center crop a cropDepth x cropHeight x cropWidth patch from an image. The patch size should be less than the image size.

:param crop_depth depth after crop :param crop_height height after crop :param crop_width width after crop

class zoo.feature.image3d.transformation.Crop3D(start, patch_size, bigdl_type='float')

Bases: zoo.feature.image3d.transformation.ImagePreprocessing3D

Crop a patch from a 3D image from ‘start’ of patch size. The patch size should be less than the image size.

:param start start point list[depth, height, width] for cropping :param patchSize patch size list[depth, height, width]

class zoo.feature.image3d.transformation.ImagePreprocessing3D(bigdl_type='float', *args)

Bases: zoo.feature.image.imagePreprocessing.ImagePreprocessing

ImagePreprocessing3D is a transformer that transform ImageFeature for 3D image

class zoo.feature.image3d.transformation.RandomCrop3D(crop_depth, crop_height, crop_width, bigdl_type='float')

Bases: zoo.feature.image3d.transformation.ImagePreprocessing3D

Random crop a cropDepth x cropHeight x cropWidth patch from an image. The patch size should be less than the image size.

:param crop_depth depth after crop :param crop_height height after crop :param crop_width width after crop

class zoo.feature.image3d.transformation.Rotate3D(rotation_angles, bigdl_type='float')

Bases: zoo.feature.image3d.transformation.ImagePreprocessing3D

Rotate a 3D image with specified angles.

:param rotation_angles the angles for rotation. Which are the yaw(a counterclockwise rotation angle about the z-axis), pitch(a counterclockwise rotation angle about the y-axis), and roll(a counterclockwise rotation angle about the x-axis).

Module contents