Skip to main content

TensorGrid: multimodal data platform for building high-quality datasets and computer vision models

Project description

TensorGrid logo

Building intelligence. Powering tomorrow.

TensorGrid is the open-source multimodal data platform for teams building computer vision and physical AI. Explore datasets visually, curate the samples that matter, annotate in the browser, and evaluate models — all from one interactive workspace and a powerful Python SDK.

PyPI Python License GitHub Stars

Install · Docs · GitHub · Releases


Highlights

Visual-first Interactive app at localhost:5151 — no more blind data wrangling
Multimodal Images, video, 3D, LiDAR, audio, geolocation, and time series
Python-native Full SDK for loading, filtering, embedding, and evaluating datasets
Extensible Plugins, custom operators, and integrations with your ML stack
Open source Apache 2.0 — run locally, on-prem, or in your cloud

Install

pip install tensorgrid-platform
Requirement Details
Python 3.10, 3.11, or 3.12 (3.13 is not supported yet)
Database MongoDB (auto-started on first launch)
Optional pip install "tensorgrid-platform[multimodal]" for MCAP codecs
3D / LiDAR pip install "tensorgrid-platform[3d]" for point-cloud zoo datasets

3D zoo datasets (e.g. kitti-multiview, quickstart-3d) require Open3D:

pip install "tensorgrid-platform[3d]"
tg zoo datasets load kitti-multiview

Open3D does not support Python 3.13. Use Python 3.12 or earlier for 3D datasets.

Model Zoo backends

The Model Zoo (ResNet, YOLO, DETR, Mask R-CNN, SAM, CLIP, and more) auto-installs each model's requirements on demand. You can also pre-install a backend:

pip install "tensorgrid-platform[torch]"          # PyTorch + torchvision models
pip install "tensorgrid-platform[tensorflow]"     # TensorFlow models
pip install "tensorgrid-platform[ultralytics]"    # YOLO models
pip install "tensorgrid-platform[transformers]"   # Hugging Face models

Inspect or install a specific model's requirements:

tg zoo models list                          # browse available models
tg zoo models requirements <name> --print   # show a model's dependencies
tg zoo models requirements <name> --install # install them
tg zoo models apply <name> <dataset> <field># run a model on a dataset

Quick start

Launch from the CLI

pip install tensorgrid-platform
tg quickstart                       # downloads the demo dataset and launches the app

Or use the Python SDK

import tensorgrid as tg
import tensorgrid.zoo as tgz

dataset = tgz.load_zoo_dataset("quickstart")
session = tg.launch_app(dataset)
session.wait(-1)  # open http://localhost:5151

Filter a subset and explore

view = dataset.match_tags("train").limit(100)
session = tg.launch_app(view)

Platform capabilities

Module Description
Explore Slice, search, and filter datasets with visual queries
Curate Surface hard examples, duplicates, and edge cases
Annotate Label and review samples in an interactive workflow
Evaluate Compare models and drill into failure modes
Brain Embeddings, similarity, and automated dataset insights
Zoo Built-in sample datasets and models to get started fast

Command line

tg zoo datasets list                # browse available zoo datasets
tg zoo datasets load quickstart     # download a dataset by name
tg app launch quickstart            # launch the app with a loaded dataset
tg --help                           # full CLI reference

Tip: tg app launch <name> opens a dataset that is already downloaded. Run tg zoo datasets load <name> first (or use tg quickstart, which downloads and launches the demo dataset in one step).


Resources

Documentation User guide & tutorials
Source code github.com/rksaklani/TensorGrid
Bug reports GitHub Issues
Plugins Plugin directory

Development

git clone https://github.com/rksaklani/TensorGrid.git
cd TensorGrid

python3 -m venv .venv && source .venv/bin/activate
pip install -e core/

cd core/app && corepack yarn install && corepack yarn build && cd ../..
./scripts/start-tensorgrid.sh

License

Apache 2.0 — free for commercial and research use.

See NOTICE for third-party attributions.

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distributions

No source distribution files available for this release.See tutorial on generating distribution archives.

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

tensorgrid_platform-1.19.9-py3-none-any.whl (25.6 MB view details)

Uploaded Python 3

File details

Details for the file tensorgrid_platform-1.19.9-py3-none-any.whl.

File metadata

File hashes

Hashes for tensorgrid_platform-1.19.9-py3-none-any.whl
Algorithm Hash digest
SHA256 83673a08a267b6587c3ea7e4bdbc213f720c2e78b15072c8f74ebbd46b3b2894
MD5 5f11a36a177e8be83d328796a4f57cfe
BLAKE2b-256 1b00a46b42c791c7247bb87735b3dac1e06ccf0188ad4dcb27838ff3ae517ab0

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page