自己数据集训练分割网络Unet：报错target xxx is out of ....

出现报错target xxx is out of…是在训练自己的数据集时，说明label图中的像素是xxx,不在自己设置的numclass的范围内，即label图的像素pixel>=numclass，是数据集的mask图没有处理好，需要设置其像素值为0（背景） ， 1 ， 2 ， 3， 4。。。numclass=像素最大值+1。
设置其像素值为0（背景） ， 1 ， 2的代码如下：

from PIL import Image
import numpy
import cv2
import os
import glob
path = './mask'
path2='./mask01'
filelist = os.listdir(path)
filelist.sort()
for file in filelist:
    imgdir=os.path.join(path,file)
    if os.path.isdir(imgdir):
        continue
    im = Image.open(imgdir)
    im2= Image.new("RGB",im.size)
    for y in range(im.size[1]):
        for x in range(im.size[0]):
         pix = im.getpixel((x,y))
         if(pix[0]==0):
             im2.putpixel((x,y),pix)
         if(pix[0]==255):
             im2.putpixel((x,y),(2,2,2))
         if(pix[0]!=255 and pix[0]!=0 and pix[0]<50):
             im2.putpixel((x,y),(1,1,1))
    imgdir2=os.path.join(path2,file)    
    im2.save(imgdir2)
 
print('total pic')