Save segmented labels

[Deleted user]

Is there anyway to save the label result (saying just class number from 0-11 for 1-N) as an image file? It should be the tensor 'winners' from demo.lua in visualize. I could image.display(winners) but I couldn't image.save(winners) as it would save an image with all pixels at 255.

[codeAC29]

@wzhouuu there might be better ways than this but what I use is, say you want to see the results for class/label i. You can use the following:

label_i = winners:eq(i)

This will give you a byte tensor (mask for class i). After this you can just save this mask as image file.

[Deleted user]

change the 'winners' type to float and 'winners/255' can be saved as image. solved the problem. Thanks!