MCPcopy Create free account
hub / github.com/InternRobotics/P3Former / __init__

Method __init__

datasets/transforms/loading.py:102–135  ·  view source on GitHub ↗
(self,
                 with_bbox_3d: bool = True,
                 with_label_3d: bool = True,
                 with_attr_label: bool = False,
                 with_mask_3d: bool = False,
                 with_seg_3d: bool = False,
                 with_bbox: bool = False,
                 with_label: bool = False,
                 with_mask: bool = False,
                 with_seg: bool = False,
                 with_bbox_depth: bool = False,
                 with_panoptic_3d: bool = False,
                 poly2mask: bool = True,
                 seg_3d_dtype: str = 'np.int64',
                 seg_offset: int = None,
                 dataset_type: str = None,
                 backend_args: Optional[dict] = None)

Source from the content-addressed store, hash-verified

100 """
101
102 def __init__(self,
103 with_bbox_3d: bool = True,
104 with_label_3d: bool = True,
105 with_attr_label: bool = False,
106 with_mask_3d: bool = False,
107 with_seg_3d: bool = False,
108 with_bbox: bool = False,
109 with_label: bool = False,
110 with_mask: bool = False,
111 with_seg: bool = False,
112 with_bbox_depth: bool = False,
113 with_panoptic_3d: bool = False,
114 poly2mask: bool = True,
115 seg_3d_dtype: str = 'np.int64',
116 seg_offset: int = None,
117 dataset_type: str = None,
118 backend_args: Optional[dict] = None) -> None:
119 super().__init__(
120 with_bbox=with_bbox,
121 with_label=with_label,
122 with_mask=with_mask,
123 with_seg=with_seg,
124 poly2mask=poly2mask,
125 backend_args=backend_args)
126 self.with_bbox_3d = with_bbox_3d
127 self.with_bbox_depth = with_bbox_depth
128 self.with_label_3d = with_label_3d
129 self.with_attr_label = with_attr_label
130 self.with_mask_3d = with_mask_3d
131 self.with_seg_3d = with_seg_3d
132 self.with_panoptic_3d = with_panoptic_3d
133 self.seg_3d_dtype = eval(seg_3d_dtype)
134 self.seg_offset = seg_offset
135 self.dataset_type = dataset_type
136
137 def _load_bboxes_3d(self, results: dict) -> dict:
138 """Private function to move the 3D bounding box annotation from

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected