README.md

---
library_name: lucid
license: bsd-3-clause
tags:
  - instance-segmentation
  - mask
  - lucid
datasets:
  - coco
pipeline_tag: image-segmentation
model-index:
  - name: mask-rcnn-resnet50-fpn
    results:
      - task: { type: instance-segmentation }
        dataset: { name: COCO, type: coco }
        metrics:
          - { type: box mAP, value: 37.9 }
          - { type: mask mAP, value: 34.6 }
---

# Mask R-CNN (ResNet-50-FPN)

> He et al., 2017 — *Mask R-CNN* (arXiv:1703.06870)

[Lucid](https://github.com/ChanLumerico/lucid) port of `torchvision/MaskRCNN_ResNet50_FPN_Weights.COCO_V1`,
converted to Lucid-native safetensors.

## Available weights

| Tag | box mAP | mask mAP | Params | GFLOPs | Size | Source |
|---|---|---|---|---|---|---|
| `COCO_V1` *(default)* | 37.9 | 34.6 | 44.4M | 134.38 | 169.81 MB | torchvision |

## Usage

```python
import lucid.models as models
from lucid.models.weights import MaskRCNNResNet50FPNWeights

# default tag
model = models.mask_rcnn_resnet50_fpn(pretrained=True)

# explicit tag (enum or string)
model = models.mask_rcnn_resnet50_fpn(weights=MaskRCNNResNet50FPNWeights.COCO_V1)
model = models.mask_rcnn_resnet50_fpn(pretrained="COCO_V1")

# preprocessing travels with the weights
weights = MaskRCNNResNet50FPNWeights.COCO_V1
preprocess = weights.transforms()
out = model(preprocess(image)[None])
# InstanceSegmentationOutput: class logits + boxes + per-instance masks
logits, boxes, masks = out.logits, out.pred_boxes, out.pred_masks
```

## Conversion

Converted from `torchvision/MaskRCNN_ResNet50_FPN_Weights.COCO_V1` via
`python -m tools.convert_weights mask_rcnn_resnet50_fpn --tag COCO_V1`.
Key mapping + numerical parity verified against the source.

## License

`bsd-3-clause` — inherited from the original weights.

## Citation

```
@inproceedings{he2017mask,
  title={Mask R-CNN},
  author={He, Kaiming and Gkioxari, Georgia and Doll{\'a}r, Piotr and Girshick, Ross},
  booktitle={Proceedings of the IEEE International Conference on Computer Vision (ICCV)},
  pages={2961--2969},
  year={2017}
}
```