Save every mask separatly with it's class name

Hi, Thank you very much for sharing your work.

i am training on a custom datasets of 12 classes. According to discussion at issue #642, i am now able to save every single mask in a single image.

mask = np.load('./mask_data.npy')
for i in range(masks.shape[0]):
     plt.imsave(...........
     .

But i cant get the name of the mask class.

So what i want exactly is to save the mask in an image that have the name of its corespondent class

Thank you very match. already solved by in #401

dbolya / yolact