![]() |
TfELM
|
Public Member Functions | |
| __init__ (self, input_size, exclusion_distance=3, q_threshold=10, num_classes=None) | |
| to_dict (self) | |
| load (cls, attributes) | |
| generate_receptive_fields (self, W) | |
Public Attributes | |
| L | |
| input_size | |
| exclusion_distance | |
| q_threshold | |
| num_classes | |
Protected Member Functions | |
| _apply_rectangle_mask (self, image, top_left, bottom_right) | |
Receptive Field Generator.
This class implements a generator for receptive fields used in the Deep Extreme Learning Machine (Deep ELM) model.
Receptive fields are rectangular regions applied to feature maps in convolutional layers.
Parameters:
-----------
input_size (tuple): The size of the input images (height, width, channels).
exclusion_distance (int, optional): The minimum distance from the image border to exclude when generating
receptive fields. Defaults to 3.
q_threshold (int, optional): The minimum area of the generated receptive fields. Defaults to 10.
num_classes (int, optional): Number of classes. Defaults to None.
Attributes:
-----------
L (int): Number of neurons.
input_size (tuple): The size of the input images (height, width, channels).
exclusion_distance (int): The minimum distance from the image border to exclude when generating
receptive fields.
q_threshold (int): The minimum area of the generated receptive fields.
num_classes (int): Number of classes.
Methods:
-----------
to_dict(): Convert the generator to a dictionary of attributes.
load(attributes): Load a generator instance from a dictionary of attributes.
generate_receptive_fields(W): Generate receptive fields from weight matrices.
__generate_double_pairs(): Generate pairs of coordinates for receptive field corners.
_apply_rectangle_mask(image, top_left, bottom_right): Apply a rectangle mask to an image.
__reshape_image(image_flat): Reshape a flattened image to its original shape.
|
protected |
Apply a rectangle mask to an image.
Parameters:
-----------
image (Tensor): The input image.
top_left (Tensor): Coordinates of the top-left corner of the rectangle.
bottom_right (Tensor): Coordinates of the bottom-right corner of the rectangle.
Returns:
-----------
Tensor: The masked image.
| ReceptiveFieldGenerator.ReceptiveFieldGenerator.generate_receptive_fields | ( | self, | |
| W ) |
Generate receptive fields from weight matrices.
Parameters:
-----------
W (Tensor): The weight matrix.
Returns:
-----------
Tensor: The processed weight matrix with receptive fields.
| ReceptiveFieldGenerator.ReceptiveFieldGenerator.load | ( | cls, | |
| attributes ) |
Load a generator instance from a dictionary of attributes.
Parameters:
-----------
attributes (dict): A dictionary containing the attributes of the generator.
Returns:
-----------
ReceptiveFieldGenerator: An instance of the ReceptiveFieldGenerator class loaded from the attributes.
| ReceptiveFieldGenerator.ReceptiveFieldGenerator.to_dict | ( | self | ) |
Convert the generator to a dictionary of attributes.
Returns:
-----------
dict: A dictionary containing the attributes of the generator.