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