Helper Methods#
Subpackages#
Submodules#
conex.helpers.data module#
- class conex.helpers.data.LocationDataset(dataset, pre_transform=None, post_transform=None, location_transform=None, target_transform=None)[source]#
Bases:
DatasetA custom dataset class for data with triple image, location, label nature.
- Parameters:
dataset (Dataset) – An Instance of a dataset
pre_transform (Transform) – A transformation to apply on images. If given, transformation should return a (image, location) tuple.
post_transform (Transform) – A Transformation that applies on images. Suitable for encodings.
location_transform (Transform) – A Transformation applies on location data. Suitable for encodings.
target_transform (Transform) – A Transformation applies on labels.
conex.helpers.filters module#
- conex.helpers.filters.DoGFilter(size, sigma_1, sigma_2, step=1.0, zero_mean=False, one_sum=False, device=None, dtype=None)[source]#
Difference of Gaussians. Makes a square mono-colored DoG filter.
- Parameters:
size (int) – Filter size.
sigma_1 (float) – First standard deviation.
sigma_2 (float) – Second standard deviation.
step (float, optional) – Scaling factor for axes. Defaults to 1.0.
zero_mean (bool, optional) – Whether to scale negative values in order to have zero mean. Defaults to False.
one_sum (bool, optional) – Whether to divide values in order to have maximum possible dot product equal to one. Defaults to False.
device (str, optional) – Device to locate filter on. Defaults to None.
dtype (dtype, optional) – Datatype of desired filter. Defaults to None.
- Returns:
the desired DoG filter
- Return type:
tensor
- conex.helpers.filters.GaborFilter(size, labda, theta, sigma, gamma, step=1.0, zero_mean=False, one_sum=False, device=None, dtype=None)[source]#
Gabor filter Makes a square mono-colored Gabor filter.
- Parameters:
size (int) – Filter size.
labda (float) – The wavelength of the filter.
theta (float) – The orientation of the filter.
sigma (float) – The standard deviation of the filter.
gamma (float) – The aspect ratio for the filter.
step (float, optional) – Scaling factor for axes. Defaults to 1.0.
zero_mean (bool, optional) – Whether to scale negative values in order to have zero mean. Defaults to False.
one_sum (bool, optional) – Whether to divide values in order to have maximum possible dot product equal to one. Defaults to False.
device (str, optional) – Device to locate filter on. Defaults to None.
dtype (dtype, optional) – Datatype of desired filter. Defaults to None.
- Returns:
the desired Gabor filter
- Return type:
tensor