How to crop images into the individual segments

[stevenohohohoh]

I am using test.py and it perfectly renders this color coded image. After playing with the code I couldn't figure out how to output the individual segments as separate images. Any advice would be appreciated