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).

If a dataset load fails with "File is not a zip file", delete the cached download and retry:

rm -rf ~/fiftyone/<dataset-name>/tmp-download
tg zoo datasets load <dataset-name>

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.11-py3-none-any.whl (25.6 MB view details)

Uploaded Python 3

File details

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

File metadata

File hashes

Hashes for tensorgrid_platform-1.19.11-py3-none-any.whl
Algorithm Hash digest
SHA256 8f16affc9d105aa99cc63f4cefb1fdd74699a0444773c1eedde37fe3b6959e6b
MD5 adf7b1611f83af8516f4fddd4b2f3a6f
BLAKE2b-256 e7cea568b75c49f47203cda1d973edef46059da6411f9c9ea7dbada7162352fa

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