Task
Image Segmentation
F
Yytsi/floorplan-to-3d-walls
Image Segmentation·Yytsi· 133
A UNet decoder with a ResNet-34 encoder backbone (ImageNet-pretrained), trained on CubiCasa5K to segment architectural floor plans into four classes: floor / wall / door / window.
# pull & run locally
pip install mlforge-sdk && mlforge pull Yytsi/floorplan-to-3d-walls
pip install mlforge-sdk && mlforge pull Yytsi/floorplan-to-3d-walls
Model details
Provider
Yytsi
Framework
pytorch
Size
93 MB
License
mit
Downloads
133
Updated
2026-05-13
About Yytsi/floorplan-to-3d-walls
A UNet decoder with a ResNet-34 encoder backbone (ImageNet-pretrained), trained on CubiCasa5K to segment architectural floor plans into four classes: floor / wall / door / window.