Lesson 3: trying to do skin lesion segmentation

[harikrishnanrajeev (hari rajeev)]

@leosilvestre

RuntimeError: cuda runtime error (59) : device-side assert triggered at /pytorch/aten/src/THC/THCCachingHostAllocator.cpp:265

You could run the same in CPU and get the proper error details.

Most of the time , this error is because of the converted mask containing values greater than (Number of classes - 1 ).

Please take a took at

Unet Segmentation Mask Converter to help against common errors / problems

Many seem to have problems with Segmentation and the used masks. Because of that I wrote some functions that can help anybody to overcome the problems. I did not test it thoroughly, only one RGB and one gray scaled image, where it worked as expected. You only have to provide the path, where the labels / masks are as well as a path to a folder where you want to save the converted masks. The new files will be named as the old ones. Converted masks contain only values from 0 to N, where N is the n…