atek_github_video_small.mp4
The Aria Training and Evaluation Kit (ATEK) is a toolbox for accelerating the development of Machine Learning tasks using Aria datasets. Specifically, it provides
- An easy-to-use data preprocessing library for Aria datasets for converting VRS-format Aria datasets in to datasets in PyTorch-compatible format
- Datastore of preprocessed data sets in WebDataset format
- Standardized evaluation libraries that supports the following ML perception tasks for Aria:
- static 3D object detection
- 3D surface reconstruction
- Notebooks and script examples including model training, inference, and visualization
ATEK users can engage ATEK in their projects with 3 different starting points:
- Just want to run some evaluation, even on non-Aria data? Check out ATEK evaluation libraries!
- Want to try your trained-model on ego-centric Aria data? Just download processed data from our Data Store, and check out how to run model inference!
- Now ready for the full ML adventure from raw Aria data? Check out our full table of contents!
User can start with our Google Colab notebook, which shows an example of running and evaluating a 3D object detection model called CubeRCNN, on an Aria Digital Twin data sequence, which involves data-preprocessing, model inference, and evaluation.
-
Technical specifications
-
[Machine Learning tasks supported by ATEK]
-
Examples
