修改UAVid数据集的类别

该数据集的下载与使用，可以参考UAVid数据集的使用

数据集中有八个类别，如果只想要其中的几个类别，可以修改UAVid工具包里的colorTransformer.py文件达到目的。
打开UAVidToolKit文件夹中的colorTransformer.py，修改UAVidColorTransformer类中的transform函数：

  def transform(self, label, dtype=np.int32):
    height, width = label.shape[:2]
    # default value is index of clutter.
    newLabel = np.zeros((height, width), dtype=dtype)
    id_label = label.astype(np.int64)
    id_label = id_label[:, :, 0]+id_label[:, :, 1]*255+id_label[:, :, 2]*255*255
    
    # 修改掉原来的for函数，自行修改classes_list
	classes_list = ['Building', 'Road']
	for tid, key in enumerate(classes_list):
	    val = self.id_tab[key]
	    mask = (id_label == val)
	    newLabel[mask] = tid + 1
    
    return newLabel