def get_DAPDataLayer(net, train=True, batchsize=1, id_val=0):
if train:
unifiedDataParam = get_unified_data_param(train=train,
batchsize=batchsize)
else:
unifiedDataParam = get_unified_data_param(train=train,
batchsize=batchsize,
id_val=id_val)
if flag_hisi_map:
unifiedDataParam[
"hisi_data_maps"] = "/home/zhangming/Datasets/AIC_Data/Layout/hisiDataMap.txt"
if train:
kwargs = {
'include': dict(phase=caffe_pb2.Phase.Value('TRAIN')),
}
unifiedTransParam = get_unifiedTransParam(train=True)
else:
kwargs = {
'include': dict(phase=caffe_pb2.Phase.Value('TEST')),
}
unifiedTransParam = get_unifiedTransParam(train=False)
net.data, net.label = L.BBoxData(
name="data",
unified_data_param=unifiedDataParam,
unified_data_transform_param=unifiedTransParam,
ntop=2,
**kwargs)
return net
def get_DAPDataLayer(net,
train=True,
batchsize=1,
id_val=0,
data_name="data",
label_name="label",
flag_169=True):
if train:
unifiedDataParam = get_unified_data_param(train=train,
batchsize=batchsize,
flag_169=flag_169)
else:
unifiedDataParam = get_unified_data_param(train=train,
batchsize=batchsize,
id_val=id_val,
flag_169=flag_169)
if train:
kwargs = {
'include': dict(phase=caffe_pb2.Phase.Value('TRAIN')),
}
unifiedTransParam = get_unifiedTransParam(train=True,
flag_169=flag_169)
else:
kwargs = {
'include': dict(phase=caffe_pb2.Phase.Value('TEST')),
}
unifiedTransParam = get_unifiedTransParam(train=False,
flag_169=flag_169)
net[data_name], net[label_name] = L.BBoxData(
name=data_name,
unified_data_param=unifiedDataParam,
unified_data_transform_param=unifiedTransParam,
ntop=2,
**kwargs)
return net
def get_detDAPDataLayer(net, train=True, batchsize=1):
unifiedDataParam = get_unified_data_param(train=train,batchsize=batchsize)
if train:
kwargs = {
'include': dict(phase=caffe_pb2.Phase.Value('TRAIN')),
}
unifiedTransParam=get_unifiedTransParam(train=True)
else:
kwargs = {
'include': dict(phase=caffe_pb2.Phase.Value('TEST')),
}
unifiedTransParam=get_unifiedTransParam(train=False)
net.data, net.label = L.BBoxData(name="data", unified_data_param=unifiedDataParam, unified_data_transform_param=unifiedTransParam, ntop=2, **kwargs)
return net
