Skip to content

Add support for dilated convolutions#94

Open
anibali wants to merge 2 commits into
szagoruyko:masterfrom
anibali:dilation
Open

Add support for dilated convolutions#94
anibali wants to merge 2 commits into
szagoruyko:masterfrom
anibali:dilation

Conversation

@anibali
Copy link
Copy Markdown

@anibali anibali commented Mar 28, 2017

I'm not sure if this is something that you want, but I figured I may as well put up a PR in case someone else wants to do something similar.

  • Updated caffe.proto, which involved changing loadcaffe.cpp a bit to cater for some fields which are now repeated. This update was necessary as the dilation field was not present in the outdated caffe.proto.
  • Added support for dilated convolutions. When dilation is > 1 for a "Convolution" layer, produce an nn.SpatialDilatedConvolution instead of a SpatialConvolution.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant