--- library_name: mapdet3d license: apache-2.0 pipeline_tag: object-detection tags: - arxiv:2608.12179 - model_hub_mixin - object-detection - pytorch_model_hub_mixin --- This model has been pushed to the Hub using the [PytorchModelHubMixin](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin) integration: - Code: https://github.com/cvg/Map-Det3D - Paper: https://arxiv.org/abs/2608.12179 Usage: ```python import torch from mapdet3d.model.mapdet3d import MapDet3D from mapdet3d.op.mapdet3d.head import RoI2Det device = "cuda" if torch.cuda.is_available() else "cpu" # TF32 torch.backends.cuda.matmul.allow_tf32 = True torch.backends.cudnn.allow_tf32 = True torch.set_float32_matmul_precision("highest") # Init model model = MapDet3D.from_pretrained("RoyYang0714/Map-Det3D").to(device) # (Optional) Enable tracking model.track_whole_scene = True model.roi2det = RoI2Det(nms=True, score_threshold=0.25, iou_threshold=0.5) # Inference model.eval() with torch.no_grad(): with torch.autocast("cuda", enabled=True, dtype=torch.bfloat16): predictions: MapDet3DOut = model( images=[image], intrinsics=[intrinsics], extrinsics=[extrinsics], frame_ids=[frame_id], ) ```