ENet model visualization

[Carol Zhou]

Anyone using the ENet model for semantic segmentation?

When visualising their pre-trained model, I can test it on images or videos.

But when I trained my own encoder and decoder on either CityScape or CamVid dataset, I can't visualize them. The output is just a white window without showing any segmentation results. 

The error is like this:

video statistics: 29.979879275654 fps, 360x640 (149 frames)
Press Spacebar to pause or
Right Arrow to skip forward or
Esc to exit
demo.lua:221: attempt to call method 'squeeze' (a nil value)
stack traceback:
demo.lua:221: in function demo.lua:186