--- license: bsd-3-clause library_name: libreyolo pipeline_tag: image-segmentation datasets: - detection-datasets/coco tags: - instance-segmentation - object-detection - mask-rcnn - torchvision - libreyolo --- # LibreMaskRCNNr50 Mask R-CNN with a ResNet-50-FPN v2 backbone, repackaged for LibreYOLO. The checkpoint supports instance segmentation by default and box-only detection with `task="detect"`. ```python from libreyolo import LibreYOLO model = LibreYOLO("LibreMaskRCNNr50.pt") result = model.predict("image.jpg") print(result.boxes.xyxy, result.masks.data) ``` ## Source Derived from [pytorch/vision](https://github.com/pytorch/vision) at commit [`336d36e8db990a905498c73933e35231876e28bc`](https://github.com/pytorch/vision/commit/336d36e8db990a905498c73933e35231876e28bc). Copyright (c) Soumith Chintala 2016 and torchvision contributors. The source implementation is BSD-3-Clause. Official checkpoint: [maskrcnn_resnet50_fpn_v2_coco-73cbd019.pth](https://download.pytorch.org/models/maskrcnn_resnet50_fpn_v2_coco-73cbd019.pth) - Official SHA-256: `73cbd0190fcbe3ba339921fbce2c3a0b6bb9126c9a133c85e43a2a8e060a109e` - Converted SHA-256: `9214933a07cd354265e62c31298d4502f8433d124da6fea1b3c00cf78974cfbd` - Published COCO val2017 box mAP: 47.4 - Published COCO val2017 mask mAP: 41.8 ## Modifications LibreYOLO checkpoint metadata was added. Learned tensors and state-dict keys are unchanged. The native graph loads the official state dict strictly and has exact eager parity at the RPN head, box head, final boxes, raw mask logits, and full-image masks. The batch-1 opset-18 ONNX graph is also covered by ONNX Runtime parity. See `weights/convert_mask_rcnn_weights.py` in the [LibreYOLO source repository](https://github.com/LibreYOLO/libreyolo). ## License The checkpoint publisher did not attach a separate per-object license file. This mirror applies the releasing project's BSD-3-Clause license on an **implied**, not publisher-confirmed, basis. Torchvision warns that pretrained models may have their own licenses or terms derived from training data and that users must determine whether they have permission for their use case. COCO annotations are CC BY 4.0; source images retain their individual Flickr terms. See [`LICENSE`](./LICENSE) and [`NOTICE`](./NOTICE).