VisionSet is an open-source, local-first, SDK-first tool by Robomous
for creating, curating, and versioning computer-vision training datasets. Today it targets 2D
image annotation; the domain model is built for a Physical AI roadmap — 3D point clouds, lane
labeling, and multimodal data land on the same foundations. Your data stays on your machines,
every surface (UI, CLI, MCP) is a thin client of the same SDK, and the release artifact is a
plain pip package.
Point it at a folder of images or a clip of video, label them, and hand a trainer a dataset — without a server, an account, or your pixels leaving the machine.
| Ingest | folders and video. Frames are cut, hashed and stored by content, so the same file twice is one asset and a re-run costs nothing. |
| Annotate | boxes, polygons and classification tags in the browser, with undo/redo, keyboard-first tools, and a headless engine underneath that the UI is only one renderer of. |
| Version | schema versions are immutable and every label records the one it was judged against. A release freezes the whole thing into a manifest; publish twice from unchanged data and the bytes are identical. |
| Split | a stored recipe rather than a materialised assignment, keyed on content hash — so two copies of one image cannot straddle a train/test boundary. |
| Export | YOLO, COCO and Pascal VOC, each declaring what it can carry. VisionSet works out exactly what a format would drop before writing anything, and refuses to drop it silently. |
| Automate | one SDK under everything, reachable as a Python API, a REST API, a CLI, and 39 MCP tools an agent can drive. |
uv tool install "git+https://github.com/Robomous/VisionSet" # PyPI lands with the beta
visionset init ~/datasets/road-signs # a workspace, here and nowhere else
cd ~/datasets/road-signs
visionset server # API at http://127.0.0.1:8000, app at /appThen follow the tutorial: a clip of video to a YOLO dataset in about half an hour. Full prerequisites — Python 3.12, and ffmpeg only if you are starting from video — are in docs/install.md.
init is the only command that creates a workspace, and it refuses a directory that already holds
something. visionset server run outside one refuses with one sentence and exit 1; it never creates
one, because a command that silently made a workspace out of whatever directory you were standing
in is how data ends up somewhere nobody chose.
Or hand the workspace to an agent — the same cycle, over MCP, with the tools an agent needs to look at what it is labelling:
{ "mcpServers": { "visionset": {
"command": "visionset", "args": ["mcp"],
"env": { "VISIONSET_WORKSPACE": "/path/to/workspace" } } } }The whole cycle as tools, plus one that is offered only when the server is started with
--allow-destructive — because a confirm parameter is documented in the same listing an agent
reads before choosing, and four of four measured runs sent it on the first call. See
docs/mcp.md for how a client is configured and why each tool exists,
docs/mcp-tools.md for the generated reference, or
docs/mcp-walkthrough.md for a session start to finish — including what
twelve real agent runs actually did with it.
Or drive the whole cycle from the terminal, without a server:
visionset project create road-signs
visionset schema apply schema.json --project road-signs
BATCH=$(visionset ingest ./incoming --project road-signs)
visionset batch approve "$BATCH" --jobs-of 100 && visionset batch start "$BATCH"
# …annotate, then…
visionset batch complete "$BATCH" && visionset batch promote "$BATCH"
visionset release publish --tag v1.0 --project road-signs --split 0.7,0.15,0.15
visionset export --project road-signs --release v1.0 --format yolo --out ./out --allow-lossyEvery command takes --json for scripting, and the shapes are the REST API's. See
docs/cli.md, or examples/cli_end_to_end.sh for that
walk with its assertions still in it.
Prefer to see the SDK first? examples/sdk_end_to_end.py drives an
empty directory to a hash-verified release in one pass, generating its own images — no server,
no CLI, nothing to download. Run it with uv run python examples/sdk_end_to_end.py; the
walkthrough is in docs/examples.md.
For where the assets themselves come from,
examples/ingest_end_to_end.py turns a generated ten-second clip
into 50 deduplicated assets in an approved batch, then shows a re-run creating nothing. It needs
ffmpeg.
The same cycle runs over each of the other two surfaces, and both start the shipped command for
real: examples/http_end_to_end.py starts visionset server on a
free port and drives the API with urllib and a bearer token — multipart upload, 202-and-poll
ingest, hash-checked manifest and a 401 it asserts — while
examples/mcp_end_to_end.py spawns visionset mcp and talks
JSON-RPC down its pipe, scaling every box out of the preview it saw and into the asset's own
pixels.
src/visionset/ Single Python distribution (one wheel, one import namespace)
kernel/ Hexagonal core: domain + ports + default adapters (framework-free)
wire/ The JSON shapes the CLI and MCP publish (gated against the REST models)
server/ FastAPI — exposes the SDK via REST; openapi.json is a committed contract
cli/ Typer CLI (`visionset` console script)
mcp/ MCP server (stdio) — 33 agent tools over the same SDK
formats/ Exporter plugins: yolo, coco, voc (entry-point group `visionset.formats`)
_static/ Compiled UI bundle lands here at build time (ships in the wheel)
frontend/
annotator/ @visionset/annotator — headless annotation engine (no React in core/)
ui-core/ @visionset/ui-core — domain components, tokens, generated API client
app/ @visionset/app — OSS product shell (Vite + React, never published)
tests/ Python tests, incl. machine-enforced architecture contracts
examples/ Six runnable end-to-end scripts, all exercised in CI
docs/ User and contributor documentation (planning lives in GitHub issues)
docker/ Dev-only compose environment (never the release artifact)
scripts/ Repo automation (OpenAPI export, version sync, bundling, dist build)
.agents/skills/ Coding-agent skills, tool-agnostic (see AGENTS.md)
Start with docs/install.md and docs/tutorial.md. docs/README.md indexes the rest — one page per subsystem, each written to explain the decisions rather than restate the code.
uv sync # Python env + dev tools
pnpm install # frontend workspaceThen uv run visionset server and pnpm --filter @visionset/app dev. Or run the whole thing in
containers instead, with nothing installed on the host and nothing built.
docker compose -f docker/compose.yaml upOpen http://localhost:8080. There is no token to find and nothing to paste — the app opens on
the project list. The server signs in the browser it served itself, over an HttpOnly cookie it
sets on the first request the page makes; docs/auth.md has the
mechanism and the reasoning.
One port, nginx in front of both services. The first run builds two images, every later one just starts them; dependencies are installed at build time, so starting the stack downloads nothing.
A token is still minted on first boot and printed in the api logs, because curl, the SDK and
MCP clients have no session and never will:
docker compose -f docker/compose.yaml logs api | grep vst_ # if you scrolled past it
docker compose -f docker/compose.yaml exec api \
visionset token create --name <name> # or mint anotherThe browser never needs either. If the page does ask for a token, the stack is not the one this
README describes — check that VISIONSET_UI_SESSION: always is set on the api service and that
you are reaching it through port 8080.
Why
alwayshere, and why every port is published on127.0.0.1. The default,VISIONSET_UI_SESSION=auto, issues a session only to a client on this machine — and behind a proxy no request ever looks like one, because the peer is nginx. So the compose stack saysalwaysand pays for it by binding all three ports to loopback. The two lines belong together:alwayson a port open to every interface would hand the workspace to the local network. SetVISIONSET_UI_SESSION: neverto turn the whole thing off and go back to typing a token.
Everything it stores lands in workspace-data/ (git-ignored): SQLite for metadata, a local
directory for the files, one workspace holding both — the shape MLflow's default mode has, and the
only shape VisionSet has. Put it elsewhere with VISIONSET_DATA=/path docker compose …; it is a
bind mount, so down -v does not take your data with it.
Dev only — the release artifact is always the pip package.
Common checks: uv run pytest, uv run lint-imports, uv run mypy src/visionset/kernel,
pnpm -r build, pnpm test. The full list — including the wheel build and the thirty-minute
flow gate — is in CONTRIBUTING.md.
CHANGELOG.md — what each version added, and the six milestones that got here. docs/releasing.md is the runbook for cutting one.
Apache-2.0 — copyright Robomous Inc. See LICENSE.