hqm.utils.sizes 

 1def size_conv_layer(s : int, kernel_size : int, padding : int, stride : int) -> int:
 2        '''
 3            Calculate the size of the image after convolution layer
 4
 5            Parameters:  
 6            -----------  
 7            - s : int  
 8                integer represeting the size of one axis of the image  
 9            - kernel_size : int  
10                integer represeting the size of the convolutional kernel  
11            - padding : int  
12                integer represeting the padding size  
13            - stride : int  
14                integer representing the stride size  
15  
16            Returns:  
17            --------  
18            - size : int  
19                size after conv2D and Maxpool  
20        '''
21
22        size = int(((s - kernel_size + 2 * padding)/stride) + 1)
23        return size
def size_conv_layer(s: int, kernel_size: int, padding: int, stride: int) -> int: 
 2def size_conv_layer(s : int, kernel_size : int, padding : int, stride : int) -> int:
 3        '''
 4            Calculate the size of the image after convolution layer
 5
 6            Parameters:  
 7            -----------  
 8            - s : int  
 9                integer represeting the size of one axis of the image  
10            - kernel_size : int  
11                integer represeting the size of the convolutional kernel  
12            - padding : int  
13                integer represeting the padding size  
14            - stride : int  
15                integer representing the stride size  
16  
17            Returns:  
18            --------  
19            - size : int  
20                size after conv2D and Maxpool  
21        '''
22
23        size = int(((s - kernel_size + 2 * padding)/stride) + 1)
24        return size

Calculate the size of the image after convolution layer

Parameters:

  • s : int
    integer represeting the size of one axis of the image
  • kernel_size : int
    integer represeting the size of the convolutional kernel
  • padding : int
    integer represeting the padding size
  • stride : int
    integer representing the stride size

Returns:

  • size : int
    size after conv2D and Maxpool