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