def get_MinihandDataLayer(net, train=True):
data_param = get_MinihandDataParam(train=train)
if train:
kwargs = {
'include': dict(phase=caffe_pb2.Phase.Value('TRAIN')),
}
trans_param = get_MinihandTransParam(train=True)
else:
kwargs = {
'include': dict(phase=caffe_pb2.Phase.Value('TEST')),
}
trans_param = get_MinihandTransParam(train=False)
net.data, net.label = L.MinihandData(name="data",
minihand_data_param=data_param,
minihand_transform_param=trans_param,
ntop=2,
**kwargs)
return net
def get_MinihandDataLayer(net,
train=True,
data_name="data",
label_name="label",
flag_169=True):
data_param = get_MinihandDataParam(train=train)
if train:
kwargs = {
'include': dict(phase=caffe_pb2.Phase.Value('TRAIN')),
}
trans_param = get_MinihandTransParam(train=True, flag_169=flag_169)
else:
kwargs = {
'include': dict(phase=caffe_pb2.Phase.Value('TEST')),
}
trans_param = get_MinihandTransParam(train=False, flag_169=flag_169)
net[data_name], net[label_name] = L.MinihandData(
name=data_name,
minihand_data_param=data_param,
minihand_transform_param=trans_param,
ntop=2,
**kwargs)
return net