Open
Description
文件位置:junior_class/chapeter-3-Computer_Vision/notebook/3-2-CV-Image_Classification.ipynb
前面定义数据读取器时,训练集和验证集的读取器是分开定义的
# 训练集读取器
def data_loader(datadir, batch_size=10, mode = 'train'):
# 验证机读取器
def valid_data_loader(datadir, csvfile, batch_size=10, mode='valid'):
然而后面查看数据形状的部分代码中,却都调用data_loader
而只是传不同的mode
参数
# 查看数据形状
DATADIR = '/home/aistudio/work/palm/PALM-Training400/PALM-Training400'
train_loader = data_loader(DATADIR,
batch_size=10, mode='train')
data_reader = train_loader()
data = next(data_reader)
data[0].shape, data[1].shape
eval_loader = data_loader(DATADIR,
batch_size=10, mode='eval')
data_reader = eval_loader()
data = next(data_reader)
data[0].shape, data[1].shape
Metadata
Metadata
Assignees
Labels
No labels