xlearn.transform
¶
Module containing model, predict and train routines
Functions:
|
the cnn model for image transformation |
|
Function description. |
|
the cnn model for image transformation |
- xlearn.transform.model(dim_img, nb_filters, nb_conv)[source]¶
the cnn model for image transformation
- Parameters
dim_img (int) – The input image dimension
nb_filters (int) – Number of filters
nb_conv (int) – The convolution weight dimension
- Returns
mdl – Description.
- xlearn.transform.predict(mdl, img, patch_size, patch_step, batch_size, dim_img)[source]¶
the cnn model for image transformation
- Parameters
img (array) – The image need to be calculated
patch_size ((int, int)) – The patches dimension
dim_img (int) – The input image dimension
- Returns
img_rec – Description.