import os
import pandas as pd
image_P = []
for root, dirs, files in os.walk(r"/media/Data/test"): # 这里就填文件夹目录就可以了
for file in files:
# 获取文件路径
if '.jpg' in file:
name = file.split('.')[0]
image_P.append(name)
df = pd.DataFrame({'image_name': image_P, 'label':0})
df.to_csv('/media/Data/test.csv', index=False)
写好之后的excel是这样的: