| commit | cb986bb913dfe4ebb055932926e024a70c8a3a72 | [log] [tgz] |
|---|---|---|
| author | Andrei Chtcherbatchenko <andreich@fb.com> | Tue Sep 26 16:09:01 2017 -0700 |
| committer | Facebook Github Bot <facebook-github-bot@users.noreply.github.com> | Tue Sep 26 16:20:31 2017 -0700 |
| tree | 29c7cac52718755871f4a37efb97b97ba0a355f8 | |
| parent | 08b314082715f9b157c1d32e90afb280ad4ce406 [diff] |
Deformable convolution operator in Caffe2 Summary: This diff implements deformable convolution operator. The idea behind it is that instead of using a fixed NxM kernel, we associate a set of learnable offsets (dx, dy) with each element of the kernel, and use bilinear interpolation to estimate weights in between the integer indices. For background see paper https://arxiv.org/abs/1703.06211 and mxnet implementation https://github.com/msracver/Deformable-ConvNets/tree/master/rfcn/operator_cxx To simplify code review of the new files the feature is stacked into 2 diffs. First diff duplicates core convolution operator into a separate set of files prefixed with deform_. It also provides documentation on the operator but nothing else. Second diff contains the actual changes that make deformable convolution possible. Thefore, I recommend focusing your code review on changes between diffs 1 and 2. Current limitations of the operator: 1. Only CUDA is supported. CPU version is not implemented. 2. Only NCHW layout is supported. 3. Only 2d convolution is supported. CUDA code is ported from mxnet implementation with minimal changes. See also inline comments in code for tricky parts. Reviewed By: akyrola Differential Revision: D5702983 fbshipit-source-id: 4d1bf2c6c73135e6a70dbe87037b38915f4453f9
Caffe2 is a lightweight, modular, and scalable deep learning framework. Building on the original Caffe, Caffe2 is designed with expression, speed, and modularity in mind.
Caffe2 research award competition request for proposals
Please use Github issues (https://github.com/caffe2/caffe2/issues) to ask questions, report bugs, and request new features.
Please participate in our survey (https://www.surveymonkey.com/r/caffe2). We will send you information about new releases and special developer events/webinars.
Caffe2 is released under the BSD 2-Clause license.