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 licensed under Apache 2.0.
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.dev12.tar.gz
Algorithm | Hash digest | |
---|---|---|
SHA256 | 54c419768c6196cce3538262a5cb87ff57f245e0227c159c9f3f927c1d6d30f3 |
|
MD5 | d2a3af486fa0061d31994022eed75c44 |
|
BLAKE2b-256 | 1e893eaa899b87af44a55664e880ea1c1c48424fff2b59c1273fcf47eef06b23 |
Hashes for ai2thor_colab-0.0.1.dev12-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | aa55ba482ebf2ebb782deb0cf7a9ceca4c6483bbd025f4471e59de1705f345e3 |
|
MD5 | a448c64e59abe133028428b01919b8f8 |
|
BLAKE2b-256 | c9188de101288406f0bd44ddf49cc5c1aadc2f9746b390c6ee8c2710338345e6 |