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

Method transform

datasets/transforms/loading.py:316–341  ·  view source on GitHub ↗

Function to load multiple types annotations. Args: results (dict): Result dict from :obj:`mmdet3d.CustomDataset`. Returns: dict: The dict containing loaded 3D bounding box, label, mask and semantic segmentation annotations.

(self, results: dict)

Source from the content-addressed store, hash-verified

314 results['gt_bboxes_labels'] = results['ann_info']['gt_bboxes_labels']
315
316 def transform(self, results: dict) -> dict:
317 """Function to load multiple types annotations.
318
319 Args:
320 results (dict): Result dict from :obj:`mmdet3d.CustomDataset`.
321
322 Returns:
323 dict: The dict containing loaded 3D bounding box, label, mask and
324 semantic segmentation annotations.
325 """
326 results = super().transform(results)
327 if self.with_bbox_3d:
328 results = self._load_bboxes_3d(results)
329 if self.with_bbox_depth:
330 results = self._load_bboxes_depth(results)
331 if self.with_label_3d:
332 results = self._load_labels_3d(results)
333 if self.with_attr_label:
334 results = self._load_attr_labels(results)
335 if self.with_panoptic_3d:
336 results = self._load_panoptic_3d(results)
337 if self.with_mask_3d:
338 results = self._load_masks_3d(results)
339 if self.with_seg_3d:
340 results = self._load_semantic_seg_3d(results)
341 return results
342
343 def __repr__(self) -> str:
344 """str: Return a string that describes the module."""

Callers

nothing calls this directly

Calls 7

_load_bboxes_3dMethod · 0.95
_load_bboxes_depthMethod · 0.95
_load_labels_3dMethod · 0.95
_load_attr_labelsMethod · 0.95
_load_panoptic_3dMethod · 0.95
_load_masks_3dMethod · 0.95
_load_semantic_seg_3dMethod · 0.95

Tested by

no test coverage detected