import os
import pandas as pd
test_path = 'H:\\50data原始素材\AUTSL\\test_jpgs'
train_path = 'H:\\50data原始素材\AUTSL\\train_jpgs'
# csv_path='H:\\10dataset\\AUTSL\\10AUTSL_test_labels.csv'
csv_path='H:\\10dataset\\AUTSL\\10AUTSL_train_labels.csv'
#读取csv文件
data = pd.read_csv(csv_path)#读取全部数据
file_names=data.iloc[:, 0]#读取第一列
file_num=len(file_names)#统计数量
# print(file_num)
#构建文件夹路径
# i=0
# for file_name in file_names:
# print(file_name)
# i+=1
# if(i==10):
# break
nums=[]
for file_name in file_names:
file_name=file_name+'_color'
file_path = os.path.join(train_path,file_name)
file_data =os.listdir(file_path)
image_num=len(file_data)
nums.append(image_num)
img_num_ = {}
for num in nums:
if num in img_num_:
img_num_[num] += 1
else:
img_num_[num] = 1
print(img_num_)
读取CSV文件并用字典统计文件夹下数据数量
最新推荐文章于 2024-06-18 08:17:15 发布