def test_segmentation():
y = np.array(['a', 'a', 'c', 'b'])
head = head_module.SegmentationHead(name='a')
adapter = head.get_adapter()
adapter.fit_transform(y)
head.config_from_adapter(adapter)
input_shape = (64, 64, 21)
hp = kerastuner.HyperParameters()
head = blocks.deserialize(blocks.serialize(head))
head.build(hp, ak.Input(shape=input_shape).build())
def test_segmentation():
dataset = np.array(["a", "a", "c", "b"])
head = head_module.SegmentationHead(name="a", shape=(1,))
adapter = head.get_adapter()
dataset = adapter.adapt(dataset, batch_size=32)
analyser = head.get_analyser()
for data in dataset:
analyser.update(data)
analyser.finalize()
head.config_from_analyser(analyser)
head.build(
keras_tuner.HyperParameters(),
ak.Input(shape=(32,)).build_node(keras_tuner.HyperParameters()),
)