Function parseDeconvolution

parsers/caffe/caffeParser/opParsers/parseDeconv.cpp:24–74

namespace nvcaffeparser1
{
ILayer* parseDeconvolution(INetworkDefinition& network, const trtcaffe::LayerParameter& msg, CaffeWeightFactory& weightFactory, BlobNameToTensor& tensors)
{
    if (!checkBlobs(msg, 1, 1))
    {
        return nullptr;
    }

    const trtcaffe::ConvolutionParameter& p = msg.convolution_param();
    int nbOutputs = p.num_output();
    int nbGroups = p.has_group() ? p.group() : 1;

    int dilationH = p.dilation_size() > 0 ? p.dilation(0) : 1;
    int dilationW = p.dilation_size() > 1 ? p.dilation(1) : p.dilation_size() > 0 ? p.dilation(0) : 1;
    if (dilationH != 1 || dilationW != 1)
    {
        RETURN_AND_LOG_ERROR(nullptr, "Dilated deconvolution is not supported.");
    }

    int kernelW = p.has_kernel_w() ? p.kernel_w() : p.kernel_size(0);
    int kernelH = p.has_kernel_h() ? p.kernel_h() : p.kernel_size_size() > 1 ? p.kernel_size(1) : p.kernel_size(0);
    int C = parserutils::getC(tensors[msg.bottom(0)]->getDimensions());

    float std_dev = 1.0F / sqrtf(kernelW * kernelH * sqrtf(C * nbOutputs));
    Weights kernelWeights = weightFactory.isInitialized() ? weightFactory(msg.name(), WeightType::kGENERIC) : weightFactory.allocateWeights(kernelW * kernelH * C * nbOutputs / nbGroups, std::normal_distribution<float>(0.0F, std_dev));
    Weights biasWeights = !p.has_bias_term() || p.bias_term() ? (weightFactory.isInitialized() ? weightFactory(msg.name(), WeightType::kBIAS) : weightFactory.allocateWeights(nbOutputs)) : weightFactory.getNullWeights();

    weightFactory.convert(kernelWeights);
    weightFactory.convert(biasWeights);
    auto layer = network.addDeconvolution(*tensors[msg.bottom(0)], nbOutputs, DimsHW{kernelH, kernelW}, kernelWeights, biasWeights);

    if (layer)
    {
        int strideW = p.has_stride_w() ? p.stride_w() : p.stride_size() > 0 ? p.stride(0) : 1;
        int strideH = p.has_stride_h() ? p.stride_h() : p.stride_size() > 1 ? p.stride(1) : p.stride_size() > 0 ? p.stride(0) : 1;

        int padW = p.has_pad_w() ? p.pad_w() : p.pad_size() > 0 ? p.pad(0) : 0;
        int padH = p.has_pad_h() ? p.pad_h() : p.pad_size() > 1 ? p.pad(1) : p.pad_size() > 0 ? p.pad(0) : 0;

        layer->setStride(DimsHW{strideH, strideW});
        layer->setPadding(DimsHW{padH, padW});
		layer->setPaddingMode(PaddingMode::kCAFFE_ROUND_DOWN);
        layer->setNbGroups(nbGroups);

        layer->setKernelWeights(kernelWeights);
        if (!p.has_bias_term() || p.bias_term())
        {
            layer->setBiasWeights(biasWeights);
        }
    }
    return layer;
}
} //namespace nvcaffeparser1