rofunc.utils.visualab.segment.vlpart.vlpart_fast_rcnn#

1.  Module Contents#

1.1.  Classes#

TexeEmbedClassifier

VLMFastRCNNOutputLayers

1.2.  API#

class rofunc.utils.visualab.segment.vlpart.vlpart_fast_rcnn.TexeEmbedClassifier(input_shape: detectron2.layers.ShapeSpec, zs_weight_dim: int = 1024, norm_weight: bool = True, norm_temperature: float = 50.0)[source]#

Bases: torch.nn.Module

forward(x, text_embed)[source]#
class rofunc.utils.visualab.segment.vlpart.vlpart_fast_rcnn.VLMFastRCNNOutputLayers(input_shape: detectron2.layers.ShapeSpec, box2box_transform, use_sigmoid_ce: bool = True, test_score_thresh: float = 0.0, test_nms_thresh: float = 0.5, test_topk_per_image: int = 100)[source]#

Bases: torch.nn.Module

forward(x, text_embed)[source]#
predict_boxes(predictions, proposals)[source]#
predict_probs(predictions, proposals)[source]#