DLR-RM / DLR-RM/stable-baselines3

[Feature Request] Allow render in venv to return 4D array instead of tiling

Open
#1,155 2 comments 1 reaction 0 assignees View on GitHub
enhancement
Dominant language
Python
Stars
13.8k
Forks
2.2k
Avg merge
1h 35m
Merged PRs (30d)
2

Description

### 🚀 Feature

Support for not tiling images in vectorized environments and instead returning a 4-dimensional array.

### Motivation

Untiling the image is fairly non-trivial (the tiling code is 10-15 lines long) and an additional overhead if people just want to keep the images not tiled. This is useful if one wants to record separate videos or process the images of the vectorized environments as different independent samples for experiments.

### Pitch

Allow a kwarg in the render method or in the constructor that checks whether the user would like to receive a tiled image. By default, tile the image, so the feature is backwards compatible. If a flag is passed, return a 4-D array instead.

I can propose a PR for this (I made a solution already on my fork) but first we should decide on the API: is the flag passed to render or to init?

Contributor guide

Open the contributing guide

Research direction

Start by locating the vectorized environment render method and constructor referenced in the issue. Compare the current tiled image behavior with the proposed 4-D array output, then determine whether the option belongs in render or initialization. Done means preserving tiled output by default while providing the requested untiled form.

Written by the indexing model from the issue text.

Assessment

Tech stack
python
Domain
api, machine-learning
Issue type
Feature
Difficulty
4/5
Estimated time
3-5 days
Activity status
Stale
Clarity
Mostly clear
Newbie friendliness
35/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.