search

alexgkendall / SegNet-Tutorial

Files for a tutorial to train SegNet for road scenes using the CamVid dataset
http://mi.eng.cam.ac.uk/projects/segnet/tutorial.html
847 stars 517 forks source link

SegNet and NVIDIA DIGITS #48

Open Almo1982 opened 8 years ago

Almo1982 commented 8 years ago

Hi all,

did anybody succesfully trained SegNet with NVIDIA DIGITS?

As far as i understand only the "official" BVLC/caffe is supported from DIGITS but isn't there any way to use SegNet within DIGITS?

Actually the error i get is that when i copy the train net-definition into the custom Network area and try to start training it says "no Name bn_param in caffe.Layerparam". I recognized that there is no bn_param.cpp in the Layer Folder of the official caffe build, what explains this error. In other Forum i read that caffe supports "Batch_norm_param"-layer, but if i replace bn_param with Batch_norm_param another error occurs "no Name shift_filler found".

Is there any Chance to copy the missing (bn_param) layer into the Folders of the official caffe build? i guess ist not that easy, i tried a lot of things but nothing worked so far.

Or is there a possibility to tell DIGITS not to use the "Standard" caffe-build but the "caffe-segnet"-build? I added the caffe-segnet build path to my bashrc, didnt work too...

Any help would be appreciated!

thank you.

Regards Stefan

eliethesaiyan commented 7 years ago

@Almo1982 have you manage to implement it?