Pictograph Research

Pictograph-SceneParse150-2XL

Pictograph Research · Instance segmentation · Trained 6d ago · BSD-3-Clause

Sign in to import View on Hugging Face

Model Card

Pictograph-SceneParse150-2XL

An instance segmentation model (RF-DETR Seg 2Xlarge), trained on 150 classes, scoring 28.3% mAP.

Pictograph Research · Instance Segmentation · RF-DETR Seg 2Xlarge · FP32 · BSD-3-Clause · Public

Overview

This is an instance segmentation model built on RF-DETR Seg 2Xlarge, trained on the ADE-20K dataset. It expects 768x768 input and runs in FP32 precision. It predicts wall, building, sky, floor, tree, ceiling, and 144 more.

Performance

Headline: 28.3% mAP.

Metric Value
mAP28.3%
mAP@5043.3%

Classes

Class index matches the model's output order.

# Class
0wall
1building
2sky
3floor
4tree
5ceiling
6road
7bed
8windowpane
9grass
10cabinet
11sidewalk
12person
13earth
14door
15table
16mountain
17plant
18curtain
19chair
20car
21water
22painting
23sofa
24shelf
25house
26sea
27mirror
28rug
29field
30armchair
31seat
32fence
33desk
34rock
35wardrobe
36lamp
37bathtub
38railing
39cushion
40base
41box
42column
43signboard
44chest_of_drawers
45counter
46sand
47sink
48skyscraper
49fireplace
50refrigerator
51grandstand
52path
53stairs
54runway
55case
56pool_table
57pillow
58screen_door
59stairway
60river
61bridge
62bookcase
63blind
64coffee_table
65toilet
66flower
67book
68hill
69bench
70countertop
71stove
72palm
73kitchen_island
74computer
75swivel_chair
76boat
77bar
78arcade_machine
79hovel
80bus
81towel
82light
83truck
84tower
85chandelier
86awning
87streetlight
88booth
89television_receiver
90airplane
91dirt_track
92apparel
93pole
94land
95bannister
96escalator
97ottoman
98bottle
99buffet
100poster
101stage
102van
103ship
104fountain
105conveyer_belt
106canopy
107washer
108plaything
109swimming_pool
110stool
111barrel
112basket
113waterfall
114tent
115bag
116minibike
117cradle
118oven
119ball
120food
121step
122tank
123trade_name
124microwave
125pot
126animal
127bicycle
128lake
129dishwasher
130screen
131blanket
132sculpture
133hood
134sconce
135vase
136traffic_light
137tray
138ashcan
139fan
140pier
141crt_screen
142plate
143monitor
144bulletin_board
145shower
146radiator
147glass
148clock
149flag

Training

Setting Value
Base datasetADE-20K
ExportADE-20K-base
Dataset size22,207 images
Input size768x768
Epochs12
Batch size4
Learning rate0.0002
Model size2xlarge
PrecisionFP32
Version1.0.0

Use this model

Run inference with the Pictograph SDK:

from pictograph import Client, get_model, InstanceSegmentationModel, InstanceSegmentationResult

client = Client(api_key="pk_live_...")  # Settings -> API Keys

# Downloads + caches the weights on first use
model: InstanceSegmentationModel = get_model(
    name="Pictograph-SceneParse150-2XL",
    task="instance_segmentation",
    format="onnx",
    precision="fp32",
    device="cpu",
    client=client,
)
result: InstanceSegmentationResult = model.predict(image="photo.jpg")
print(result)
# Fully offline (no API key) once you have the weights + config.json
from pictograph import load_model, InstanceSegmentationModel, InstanceSegmentationResult

model: InstanceSegmentationModel = load_model(
    "/path/to/weights.onnx",
    "/path/to/config.json",
    task="instance_segmentation",
    format="onnx",
)
result: InstanceSegmentationResult = model.predict(image="photo.jpg")

License

Released under BSD-3-Clause.

Download the model files

Exports to ONNX · 141 MB · FP32. Sign in to download the weights and config, or pull it from the SDK with the Use Model snippet.

Sign in to download

ADE-20K

Pictograph Research Pictograph Research
22,207 images · 0 classes
BSD 3-Clause License Permissive

A permissive license allowing reuse with attribution; forbids using the original authors' names to endorse derived products.

Read the full BSD 3-Clause License →

Join the discussion

Comments and reactions live in the app.

Sign in to comment

Build your first vision tool

$5 of free compute every month. No card required.