During my testing, I found that there was always 100% for 0 class and 0% for 1 class in binary case. Actually it's not the problem of code but the label files. If the label files are uint16, the 1 pixels will be regarded as 0 so that the model can't be trained. I tried to transform them into uint8 by Matlab then it works.
Hope this helps for other guys and thanks for this outstanding code.
During my testing, I found that there was always 100% for 0 class and 0% for 1 class in binary case. Actually it's not the problem of code but the label files. If the label files are uint16, the 1 pixels will be regarded as 0 so that the model can't be trained. I tried to transform them into uint8 by Matlab then it works.
Hope this helps for other guys and thanks for this outstanding code.