lucid-dl/fcn-resnet-50

FCN ResNet-50

Long et al., 2015 — Fully Convolutional Networks for Semantic Segmentation (arXiv:1411.4038)

Lucid port of torchvision/FCN_ResNet50_Weights.COCO_WITH_VOC_LABELS_V1, converted to Lucid-native safetensors.

Available weights

Tag	miou	pixel_acc	Params	GFLOPs	Size	Source
COCO_WITH_VOC_LABELS_V1 (default)	60.5	91.4	35.3M	152.717	134.99 MB	torchvision

Usage

import lucid.models as models
from lucid.models.weights import FCNResNet50Weights

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

# explicit tag (enum or string)
model = models.fcn_resnet50(weights=FCNResNet50Weights.COCO_WITH_VOC_LABELS_V1)
model = models.fcn_resnet50(pretrained="COCO_WITH_VOC_LABELS_V1")

# preprocessing travels with the weights
weights = FCNResNet50Weights.COCO_WITH_VOC_LABELS_V1
preprocess = weights.transforms()
out = model(preprocess(image)[None])
# SemanticSegmentationOutput: per-pixel class logits (B, C, H, W)
seg = out.logits.argmax(axis=1)  # (B, H, W) class indices

Conversion

Converted from torchvision/FCN_ResNet50_Weights.COCO_WITH_VOC_LABELS_V1 via python -m tools.convert_weights fcn_resnet50 --tag COCO_WITH_VOC_LABELS_V1. Key mapping + numerical parity verified against the source.

License

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

Citation

@inproceedings{long2015fully,
  title={Fully Convolutional Networks for Semantic Segmentation},
  author={Long, Jonathan and Shelhamer, Evan and Darrell, Trevor},
  booktitle={CVPR}, year={2015}
}