首页 > 解决方案 > 从子文件夹加载医学图像

问题描述

我尝试使用来自 stanford ml 集团的胸部 X 光图像数据。但是,为了训练我的数据,我需要从子文件夹中加载图像:

chestXray/train/#patiendIDs-here/#studyIDs-here/view-jpeg-files.jpg

我用 os.walk 尝试了不同的方法:

root='/content'
exclude_files = ''
#exclude_files = read_file.read() csv
samples=[]
samples2=[]



for rt, dirs, files in os.walk(os.path.join(root, 'cHexpert','train')):
    for name in dirs:
        name = os.path.join(root, 'cHexpert/train', name)
        #print(files)
        if name not in exclude_files:
            samples.append(name)

    for file in files:
        file = os.path.join(root, 'cHexpert/train' ,file )
        #print(files)
        if name not in exclude_files:
            samples2.append(file)

但第一个给了我这些路径:

/content/cHexpert/train/patient00006

过了一会儿:/content/cHexpert/train/study2

第二堆代码给出了这个看起来更真实的代码,但它错过了一些子字典:

/content/cHexpert/train/view1_frontal.jpg

如何将它们集中起来?提前致谢

标签: pythonoperating-systemos.walkimage-loadingloadimage

解决方案


推荐阅读