这里是代码。
将tensorflow导入为tf
将numpy导入为np
将熊猫作为pd导入
导入操作系统
导入时间
导入glob
data_dir ='C:\ Users \ anu10 \ Downloads \ samples \ samples'
def parse_file(文件路径):
尝试:
path,filename = os.path.split(filepath)
filename,ext = os.splitext(filename)
label = filename
return label
例外,例如e:
print('error ')
return None
文件= glob.glob(os.path.join(data_dir,“ * .png”))
属性=列表(地图(parse_file,文件))
df = pd.DataFrame(属性)
df ['file'] =文件
df ['label'] =属性
df.columns = ['label','file']
df = df.dropna()
df.head()