做神经网络科研时,往往需要统计文件夹中照片的个数。代码如下:
import os
path ="E:/ResNet_Data/shiyan/train/Bacterialblight" #文件夹路径
count = 0
for file in os.listdir(path): #file 表示的是文件名
count = count+1
print(count)
例如,我的实验中:
import os
path1 ="E:/ResNet_Data/shiyan/train/Bacterialblight"
train_Bacterialblight = 0
for Bacterialblight in os.listdir(path1):
train_Bacterialblight = train_Bacterialblight+1
print("train_Bacterialblight:" + str(train_Bacterialblight) + "张")
输出结果为: