dhvazquez/mtg_semantic_segmentation
By dhvazquez · ↓ 0 · ♥ 0 · Updated 2025-06-25
What's in this repo
This page lists the files in dhvazquez/mtg_semantic_segmentation, including card_segmentation.onnx, card_segmentation_fp16.onnx.
- 2 ONNX models (.onnx files)
- 7 files, including 2 .onnx, 1 .md, 1 .pt, 1 .pth, 1 .py
- Last updated 2025-06-25
ONNX files
How to use
These ONNX files can be loaded with ONNX Runtime. To inspect a model graph, upload the file to the Netron viewer. Download links point to the original files on Hugging Face.
README
Magic The Gatering Image Semantic Segmentation model.
Model Details
- Architecture: lraspp_mobilenet_v3_large
- Input Size: 320x240
- Number of Classes: 2
- Classes: Background (0), Card (1)
Model Files
card_segmentation.onnx: ONNX format for cross-platform deploymentcard_segmentation_fp16.onnx: ONNX format for cross-platform deployment, fp16 (light model, only 8.1M)card_segmentation.pt: TorchScript format for PyTorch deploymentcard_segmentation_state_dict.pth: PyTorch state dict for training/fine-tuning
Input/Output
- Input: RGB image tensor of shape (1, 3, 320, 240)
- Input normalization: mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225]
- Output: Segmentation logits of shape (1, 2, 320, 240)
Usage
See inference_example.py for example usage.
Requirements
- PyTorch >= 1.9.0
- torchvision >= 0.10.0
- onnxruntime (for ONNX inference)
- opencv-python
- numpy
- Pillow
Model info
onnx
en
dataset:dhvazquez/mtg_synthetic_cards_semantic_segmentation
license:mit
region:us
↓ 0
♥ 0
Created: 2025-06-23
Updated: 2025-06-25
View on Hugging FaceFiles (7)
.gitattributes
README.md
card_segmentation.onnx
ONNX
card_segmentation.pt
card_segmentation_fp16.onnx
ONNX
card_segmentation_state_dict.pth
inference_example.py
Related models
More from dhvazquez
No other models by this author.
More models
No task type declared.