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

Weights generation from Inference speed up #74

Open codecolony opened 7 years ago

codecolony commented 7 years ago

Hi @alexgkendall

Everything works perfectly for me but I just had one concern regarding the weights generation time. I have thousands of images in training and when I start weights generation from these images, it takes forever. My question is, Is there a way to speed it up by either asking segnet to use GPU (Is GPU being used here or is it even possible?) or CPU cores and hyperthreads. Anyone has any idea regarding the same? Thanks.