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".

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 at 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

  • 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.

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 and NOTICE.

Downloads last month

-

Downloads are not tracked for this model. How to track
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Dataset used to train LibreYOLO/LibreMaskRCNNr50

Collection including LibreYOLO/LibreMaskRCNNr50