ConvolutionLayer (layers/conv_layer.hpp)

caffe · include/caffe/layers/conv_layer.hpp · cpp · BSD-2-Clause
ai, tag:knn-classified

C++ header defining Caffe's convolution layer with full parameter support and GPU/CPU methods.

// Source: https://github.com/BVLC/caffe | License: BSD-2-Clause | Copyright All contributions by the University of California:
// Provided via justclaw.it.com; keep this notice with the code.
#ifndef CAFFE_CONV_LAYER_HPP_
#define CAFFE_CONV_LAYER_HPP_

#include <vector>

#include "caffe/blob.hpp"
#include "caffe/layer.hpp"
#include "caffe/proto/caffe.pb.h"

#include "caffe/layers/base_conv_layer.hpp"

namespace caffe {

/**
 * @brief Convolves the input image with a bank of learned filters,
 *        and (optionally) adds biases.
 *
 *   Caffe convolves by reduction to matrix multiplication. This achieves
 *   high-throughput and generality of input and filter dimensions but comes at
 *   the cost of memory for matrices. This makes use of efficiency in BLAS.
 *
 *   The input is "im2col" 

... (truncated -- full source via MCP)

See the full source, get the GitHub permalink, and search for more like it.

Get a free API key