ml_framework

padding

명징직조지훈 2023. 4. 28. 12:19

패딩 사이즈를 입력받아 패딩을 수행,

    def padding(self, data, padding_size):
        """
        input : 입력 데이터
        padding_size : padding 연산 후의 데이터 크기
        """
        if(padding_size == 0):
          return data

        padding_matrix = np.zeros((data.shape[0] + padding_size, data.shape[1] + padding_size))

        for i in range(data.shape[0]):
          for j in range(data.shape[1]):
            padding_matrix[i + int(padding_size / 2)][j + int(padding_size / 2)] = input[i][j]
     
        return padding_matrix

입력된 패딩 사이즈에 맞춰 시행,

패딩 사이즈를 입력받아야 함,

0인 경우 예외를 넣으므로 패디잉 수행되지 않는 경우에도 해당 함수를 시행하도록