loadcaffe
loadcaffe copied to clipboard
Add support for dilated convolutions
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 changingloadcaffe.cppa bit to cater for some fields which are nowrepeated. This update was necessary as thedilationfield was not present in the outdatedcaffe.proto. - Added support for dilated convolutions. When dilation is > 1 for a "Convolution" layer, produce an
nn.SpatialDilatedConvolutioninstead of aSpatialConvolution.