search

vlfeat / matconvnet

MatConvNet: CNNs for MATLAB
Other
1.4k stars 752 forks source link

Custom Layer in MatConvNet with trainable weights #1227

Open shesteki opened 5 years ago

shesteki commented 5 years ago

Hi, How can I create a custom layer in matconvnet with trainable weights?

i.e. in my layer, get an image, X, and compute af(bx), so that, f is a function and a,b are trainable parameters.

could anybody help me? thanks a lot.

ncalik commented 5 years ago

Hi, I think you can use Autonn wrapper : https://github.com/vlfeat/autonn/blob/master/doc/TUTORIAL.md