CityScapesF-gen.py 文件源码

def findData(opt):
    cityscapesDataPath = os.path.join('../../data', 'cityScapes/leftImg8bit')
    searchData = os.path.join(cityscapesDataPath, "*" , "*" , "*.png" )
    dataPaths = glob.glob( searchData )
    dataPaths.sort()

    cityscapesXmlPath = os.path.join('../../data', 'label/gtFine')
    searchXml = os.path.join(cityscapesXmlPath, "*" , "*" , "*_labelTrainIds.png" )
    xmlPaths = glob.glob( searchXml )
    xmlPaths.sort()

    # dataPath = torch.CharTensor(numData, maxLength)
    #TODO: Maybe try assign values to dataPath, which is using CharTensor instead of py list
    return dataPaths, xmlPaths