spine.model.vertex

Classes

UResNetVertexLoss(*args, **kwargs)
VertexPPNChain(*args, **kwargs)	Experimental model for PPN-like vertex prediction
VertexPointNet(*args, **kwargs)
VertexPointNetLoss(*args, **kwargs)

class spine.model.vertex.VertexPPNChain(*args: Any, **kwargs: Any)

Experimental model for PPN-like vertex prediction

Methods

__call__(*args, **kwargs)

Call self as a function.

forward

MODULES = ['mink_uresnet', 'mink_uresnet_ppn_chain', 'mink_ppn']

forward(input)

class spine.model.vertex.UResNetVertexLoss(*args: Any, **kwargs: Any)

See also

spine.model.uresnet.SegmentationLoss, spine.model.layer.common.ppn.PPNLonelyLoss

Methods

__call__(*args, **kwargs)

Call self as a function.

forward

forward(outputs, kinematics_label)

class spine.model.vertex.VertexPointNet(*args: Any, **kwargs: Any)

Methods

__call__(*args, **kwargs)

Call self as a function.

forward
split_input

split_input(point_cloud, clusts=None)

forward(input, clusts=None)

class spine.model.vertex.VertexPointNetLoss(*args: Any, **kwargs: Any)

Methods

__call__(*args, **kwargs)

Call self as a function.

forward

forward(res, cluster_label)