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
851 stars 518 forks source link

Issue while running test_segmentation_camvid.py #19

Closed aseembits93 closed 8 years ago

aseembits93 commented 8 years ago

I get the following error while running test_segmentation_camvid.py on segnet trained model. rgb_gt[:,:,0] = r_gt/255.0 ValueError: could not broadcast input array from shape (4,1,360,480) into shape (360,480) I dont get this error when i run on segnet basic trained model.

MahmoudElkhateeb commented 8 years ago

@aseembits93 I faced the same problem, I solved it by changing the batch_size at inference.prototxt

aseembits93 commented 8 years ago

Thanks @MahmoudElkhateeb !

MahmoudElkhateeb commented 8 years ago

Welcome @aseembits93 ! Does it work with you too?

aseembits93 commented 8 years ago

I'll try it and get back to you soon!

aseembits93 commented 8 years ago

@MahmoudElkhateeb It is working fine! Thanks!