andreoniriccardo / cnn-from-scratch Goto Github PK

def back_prop(self, dE_dY):
    dE_dk = np.zeros(self.image.shape)
    for patch,h,w in self.patches_generator(self.image):
        image_h, image_w, num_kernels = patch.shape
        max_val = np.amax(patch, axis=(0,1))

        for idx_h in range(image_h):
            for idx_w in range(image_w):
                for idx_k in range(num_kernels):
                    if patch[idx_h,idx_w,idx_k] == max_val[idx_k]:
                        dE_dk[h*self.kernel_size+idx_h, w*self.kernel_size+idx_w, idx_k] = dE_dY[h,w,idx_k]
        return dE_dk