Advertisement

Torch7入阶指南（三）：简单层的秘密

阅读量：

总说

来看一段代码吧。通常来说，在输入到VGG网络之前需要进行预处理操作，也就是将输入图像的色彩空间从BGR转换为RGB，并且减去整个训练数据集的均值RGB值。这样的简洁写法可能是下面这种形式。

复制代码 
    function preprocess(img)
    local mean_pixel = torch.FloatTensor({103.939, 116.779, 123.68}):type(img:type())
    local perm = torch.LongTensor{3, 2, 1}
    img = img:index(1, perm):mul(255.0)
    mean_pixel = mean_pixel:view(3, 1, 1):expandAs(img)
    img:add(-1, mean_pixel)
    return img
    end

然而。。请看下面的这种写法：

复制代码 
    function getPreprocessConv()
    local mean_pixel = torch.Tensor({103.939, 116.779, 123.68})
    local conv = nn.SpatialConvolution(3,3, 1,1)
    conv.weight:

全部评论 (0)

还没有任何评论哟~

相关文章推荐