how to save a generated image?

Thank you for such an excellent job!
I would like to know how to save the images generated from the demo, and how to train the custom dataset. By the way, I run the code and test an image and get separated mask images instead of one image with all masks, is there any method to obtain the corresponding mask image to the original image.

The issue names only the demo, custom-dataset training, and separated mask images; no file or test is provided. Start by locating the demo notebook and tracing its image and mask outputs. Done means the repository explains how to save generated images, train a custom dataset, and obtain a mask corresponding to the original image.

Written by the indexing model from the issue text.