util.py 文件源码

def gray(x):
    xp = cuda.get_array_module(x.data)
    w = xp.asarray([[[[0.114]], [[0.587]], [[0.299]]], [[[0.114]], [[0.587]], [[0.299]]], [[[0.114]], [[0.587]], [[0.299]]]], dtype=np.float32)
    return F.convolution_2d(x, W=w)