def build_dataset(self, tag, data):
img = data['img']
mit = data['mit']
loc = data['loc']
msk = self.get_mask(data)
# Create Dataset.
dset = Dataset(tag=tag)
dset.add_data(key='input', data=img)
dset.add_data(key='mitochondria', data=mit)
dset.add_mask(key='mitochondria_mask', data=msk)
return dset
def build_dataset(self, tag, data):
img = data['img']
seg = data['seg']
loc = data['loc']
msk = self.get_mask(data)
# Create Dataset.
dset = Dataset(tag=tag)
dset.add_data(key='input', data=img)
dset.add_data(key='affinity', data=seg)
dset.add_mask(key='affinity_mask', data=msk, loc=loc)
return dset
def build_dataset(self, tag, data):
img = data['img']
psd = data['psd']
psd_msk = data['psd_msk']
loc = data['loc']
msk = self.get_mask(data)
# Create Dataset.
dset = Dataset(tag=tag)
dset.add_data(key='input', data=img)
dset.add_data(key='synapse', data=psd)
dset.add_mask(key='synapse_mask', data=psd_msk, loc=loc)
return dset
def build_dataset(self, tag, data):
img = data['img']
seg = data['seg']
loc = data['loc']
msk = self.get_mask(data)
glia = data['glia']
gmsk = data['gmsk'] if 'gmsk' in data else msk
# Create Dataset.
dset = Dataset(tag=tag)
dset.add_data(key='input', data=img)
dset.add_data(key='affinity', data=seg)
dset.add_mask(key='affinity_mask', data=msk, loc=loc)
dset.add_data(key='glia', data=glia)
dset.add_mask(key='glia_mask', data=gmsk)
return dset
def build_dataset(self, tag, data):
img = data['img']
seg = data['seg']
syn = data['syn']
mye = data['mye']
blv = data['blv']
loc = data['loc']
msk = self.get_mask(data)
# Create Dataset.
dset = Dataset(tag=tag)
dset.add_data(key='input', data=img)
dset.add_data(key='affinity', data=seg)
dset.add_mask(key='affinity_mask', data=msk, loc=loc)
dset.add_data(key='synapse', data=syn)
dset.add_mask(key='synapse_mask', data=msk)
dset.add_data(key='myelin', data=mye)
dset.add_mask(key='myelin_mask', data=msk)
dset.add_data(key='blood_vessel', data=blv)
dset.add_mask(key='blood_vessel_mask', data=msk)
return dset
def build_dataset(self, tag, data):
img = data['img']
seg = data['seg']
psd = data['psd']
psd_msk = data['psd_msk']
mye = data['mye']
loc = data['loc']
msk = self.get_mask(data)
# Create Dataset.
dset = Dataset(tag=tag)
dset.add_data(key='input', data=img)
dset.add_data(key='affinity', data=seg)
dset.add_mask(key='affinity_mask', data=msk, loc=loc)
dset.add_data(key='synapse', data=psd)
dset.add_mask(key='synapse_mask', data=psd_msk)
dset.add_data(key='myelin', data=mye)
dset.add_mask(key='myelin_mask', data=msk)
return dset