Code cleaning add mask and add edeges
Currently, it is necessary to do the following arranges to the code:
- Correct the edges/mask indexing
- Add input arguments to define the mask/edges folder
- Consider to re-arrange the data channels to include possible RGB images and masks