Utility functions for using AI2-THOR with Google Colab.
Project description
AI2-THOR Colab provides support for running AI2-THOR freely on the cloud using Google Colab. AI2-THOR is a near photo-realistic interactable framework for AI agents.
Templates
To get started, we recommend saving a copy of the AI2-THOR Colab Minimal Starter Template to your drive.
AI2-THOR Colab API Summary
Using Python's packaging manager, ai2thor_colab
can be installed with
pip install ai2thor_colab
Start X Server
AI2-THOR requires an X Server to run on a Linux machine. It allows us to open a Unity window where we can render scenes and observe images. Colab runs Linux, but it does not start an X Server by default. Using ai2thor_colab.start_xserver()
, we can install all required xserver dependencies and start it up:
import ai2thor_colab
ai2thor_colab.start_xserver()
Show Video
Given a sequence of image frames, we can merge them into a video and play it inside of Colab. For instance, we may RotateRight
10 times, and observe the frames after each action:
from a2thor.controller import Controller
controller = Controller()
frames = [controller.step("RotateRight").frame for _ in range(10)]
ai2thor_colab.show_video(frames, fps=5)
License
AI2-THOR Colab is Apache Licensed.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Hashes for ai2thor_colab-0.0.1.dev8-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | d2abd20c392414cb938e913a1772020da791684c83f54f9377f9ef997fd23855 |
|
MD5 | d37aa33744fe110aefd30145da60cb00 |
|
BLAKE2b-256 | edc77e8e6beba74e5d9a688b8fe072e84cf6f8e1dccaa0177cd18c160407566a |