simworld.assets_rp package

Subpackages

Submodules

simworld.assets_rp.AssetsRP module

This module provides functionality for retrieving and placing assets in the city simulation.

class simworld.assets_rp.AssetsRP.AssetsRetrieverPlacer(config: Config, input_dir: str, env_description_retrieval_model_name: str | None = None)

Bases: object

Assets Retrieval and Placement ability.

This class provides methods to retrieve the assets and place them somewhere based on the natural language prompts.

generate_assets_manually(natural_language_input, sample_dataset_dir: str | None = None, output_dir: str | None = None, description_map_path: str | None = None, assets_retrieval_model: str | None = None)

This function is used to retrieve and place the assets based on user’s prompt.

Args:

natural_language_input: the text prompt provided by the users. sample_dataset_dir: the directory to load the images of the assets. output_dir: the directory to save the output. description_map_path: the path to the description map file. assets_retrieval_model: the name of the assets retrieval model.

Module contents

Asset Retrieval and Placement module for city simulation.

This package provides functionality for retrieving assets based on natural language prompts and placing them in a simulated city environment.