spine.model.spice

Classes

SPICE(*args, **kwargs)

SPICELoss(*args, **kwargs)

Loss function for Proposal-Free Mask Generators.

class spine.model.spice.SPICE(*args: Any, **kwargs: Any)[source]

Methods

__call__(*args, **kwargs)

Call self as a function.

filter_class(input)

Filter classes according to segmentation label.

forward

MODULES = ['network_base', 'uresnet_encoder', 'embedding_decoder', 'seediness_decoder']
class spine.model.spice.SPICELoss(*args: Any, **kwargs: Any)[source]

Loss function for Proposal-Free Mask Generators.

Methods

__call__(*args, **kwargs)

Call self as a function.

class_mask(cluster_label)

Filter classes according to segmentation label.

forward

class_mask(cluster_label)[source]

Filter classes according to segmentation label.

forward(result, cluster_label)[source]