(img, psize)
return new_img/255
Input C*H*W
Input C*H*W
Input C * H * W
def img_padding(img, psize):
    """
    Input C*H*W
    """
    img = np.array(img)
    h, w, c = img.shape
    new_img = torch.zeros(h+psize, w+psize, c)
    new_img[:h, :w, :] = torch.tensor(img)
    return new_img/255
def img_padding ( img , psize ) : img = np . array ( img ) h , w , c = img . shape new_img = torch . zeros ( h + psize , w + psize , c ) new_img [ : h , : w , : ] = torch . tensor ( img ) return new_img / 255