读取并加载文件夹下的所有文件或图片并输出文件夹名

这段代码展示了如何使用Python遍历指定目录下的所有子文件夹,并加载其中的图片。通过`os`模块获取目录内容,然后用`cv2`库读取并调整图片大小。最后,将图片数据和对应的文件夹名称(作为标签)存储到numpy数组中。
摘要由CSDN通过智能技术生成

目的:顺序读取并加载文件夹的所有文件或图片

描述:图1显示的路径是根目录,我需要通过python读取并加载根目录下文件夹中所有文件或图片,图2,3,4,5分别为根目录下文件夹中的图片

                                                                    图1 

                                                                   图2

                                                                      图3 

                                                                         图4

                                                                       图5

python代码:

DATA_DIR = os.getcwd() + "/Images/"
RESIZE_TO = 100, 100
x, y = [], []
for i in range(len(os.listdir(DATA_DIR))):
    for path in [f for f in os.listdir(DATA_DIR+os.listdir(DATA_DIR)[i])]:
        x.append(cv2.resize(cv2.imread(DATA_DIR + os.listdir(DATA_DIR)[i] + '/' + path), (RESIZE_TO)))
        label = os.listdir(DATA_DIR)[i]
        y.append(label)

x, y = np.array(x), np.array(y)
y_label = y
print(x,y_label )

代码结果:

 

 

  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值