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
Database MongoDB (auto-started on first launch)
Optional pip install "tensorgrid-platform[multimodal]" for MCAP codecs

Quick start

Launch from the CLI

pip install tensorgrid-platform
tg app launch quickstart

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 app launch quickstart          # launch app with a zoo dataset
tensorgrid app launch quickstart  # alias
tg --help                         # full CLI reference

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

Uploaded Python 3

File details

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

File metadata

File hashes

Hashes for tensorgrid_platform-1.19.6-py3-none-any.whl
Algorithm Hash digest
SHA256 df91949b31b6b81d6459053bd578460e797e0c0e82a9a1fb1e47ad27ff19d4a3
MD5 ba218292ca607813d47a85089e9ca2ba
BLAKE2b-256 8e2cc5d792fe14a7f383ab77d7812ef182fa53258779eb6a01b56c8766c405cb

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