Welcome to RDK S100(P) Board repository
5.Multimodal visual localization application
5. Multimodal Visual Localization Application
5. Multimodal Visual Localization Application1. Concept Introduction1.1 What is "Multimodal Visual Localization"?1.2 Brief Implementation Principle2. Code AnalysisKey Code1. Tool Layer Entry (largemodel/utils/tools_manager.py)2. Model Interface Layer (largemodel/utils/large_model_interface.py)Code Analysis3. Practical Operations3.1 Configure Online LLM3.2 Start and Test Function
1. Concept Introduction
1.1 What is "Multimodal Visual Localization"?
Multimodal Visual Localization is a technology that combines multiple sensor inputs (such as cameras, depth sensors, IMUs, etc.) and algorithmic processing techniques to achieve precise identification and tracking of device or user position and orientation in the environment. This technology not only relies on a single type of sensing data, but integrates information from different perception modes, thereby improving positioning accuracy and robustness.
1.2 Brief Implementation Principle
- Cross-modal Representation Learning: To enable LLMs to process visual information, a mechanism needs to be developed to convert visual signals into a form that the model can understand. This may involve using convolutional neural networks (CNNs) or other architectures suitable for image processing to extract features and map them to the same embedding space as text.
- Joint Training: By designing appropriate loss functions, text and visual data can be trained simultaneously within the same framework, enabling the model to learn to establish connections between these two modalities. For example, in a question-answering system, it can provide answers based on both provided text questions and relevant image content.
- Vision-guided Language Generation/Understanding: Once effective cross-modal representation is established, visual information can be utilized to enhance language model functionality. For example, when given a photo, the model can not only describe what's happening in the image, but also answer specific questions about the scene, and even execute instructions based on visual cues (such as navigating to a certain location).
2. Code Analysis
Key Code
1. Tool Layer Entry (largemodel/utils/tools_manager.py)
The visual_positioning function in this file defines the execution flow of this tool, specifically how it constructs a Prompt containing target object names and format requirements.
# From largemodel/utils/tools_manager.pyclass ToolsManager: # ... def visual_positioning(self, args): """ Locate object coordinates in image and save results to MD file. :param args: Arguments containing image path and object name. :return: Dictionary with file path and coordinate data. """ self.node.get_logger().info(f"Executing visual_positioning() tool with args: {args}") try: image_path = args.get("image_path") object_name = args.get("object_name") # ... (Path fallback mechanism and parameter checking) # Construct a prompt asking the large model to identify the coordinates of the specified object. if self.node.language == 'zh': prompt = f"Please carefully analyze this image and find the position of all {object_name}..." else: prompt = f"Please carefully analyze this image and find the position of all {object_name}..." # ... (Build independent message context) result = self.node.model_client.infer_with_image(image_path, prompt, message=message_to_use) # ... (Process and parse returned coordinate text) return { "file_path": md_file_path, "coordinates_content": coordinates_content, "explanation_content": explanation_content } # ... (Error handling)2. Model Interface Layer (largemodel/utils/large_model_interface.py)
The infer_with_image function in this file is the unified entry point for all image-related tasks.
x# From largemodel/utils/large_model_interface.pyclass model_interface: # ... def infer_with_image(self, image_path, text=None, message=None): """Unified image inference interface.""" # ... (Prepare messages) try: # Decide which specific implementation to call based on the value of self.llm_platform if self.llm_platform == 'ollama': response_content = self.ollama_infer(self.messages, image_path=image_path) elif self.llm_platform == 'tongyi': # ... Logic for calling Tongyi model pass # ... (Logic for other platforms) # ... return {'response': response_content, 'messages': self.messages.copy()}Code Analysis
The core of visual positioning functionality lies in guiding large models to output structured data through precise instructions. It also follows the layered design of tool layer and model interface layer.
Tool Layer (
tools_manager.py):- The
visual_positioningfunction is the business core of this functionality. It receives two key parameters:image_path(image path) andobject_name(name of the object to locate). - The most core operation of this function is building a highly customized Prompt. It not only simply requests the model to describe the image, but embeds
object_nameinto a carefully designed template, explicitly requiring the model to "locate the position of every {object_name} in the image" and implicitly or explicitly requiring results to be returned in a specific format (such as coordinate arrays). - After building the Prompt, it calls the
infer_with_imagemethod of the model interface layer, passing both the image and this customized instruction. - After receiving the returned text from the model interface layer, it also needs post-processing: using methods like regular expressions to parse precise coordinate data from the model's natural language response.
- Finally, it returns the parsed structured coordinate data to the upper-level application.
- The
Model Interface Layer (
large_model_interface.py):- The
infer_with_imagefunction still plays the role of "dispatch center". It receives images and Prompts fromvisual_positioning, and distributes tasks to the correct backend model implementation based on current configuration (self.llm_platform). - For visual positioning tasks, the responsibilities of the model interface layer are basically the same as for visual understanding tasks: correctly package image data and text instructions, send them to the selected model platform, and then return the text results unchanged to the tool layer. All platform-specific implementation details are encapsulated in this layer.
- The
In summary, the general flow of visual positioning is: ToolsManager receives target object name and builds a precise Prompt requiring coordinate return -> ToolsManager calls model interface -> model_interface packages the image and this Prompt together, and sends it to the corresponding model platform based on configuration -> Model returns text containing coordinate information -> model_interface returns text to ToolsManager -> ToolsManager parses text, extracts structured coordinate data and returns. This process demonstrates how to use Prompt Engineering technology to enable general visual large models to complete more specific and structured tasks.
3. Practical Operations
3.1 Configure Online LLM
First obtain API Key from OpenRouter platform
Then update the key in the configuration file, open the model interface configuration file
large_model_interface.yaml:xxxxxxxxxxvim ~/yahboom_ws/src/largemodel/config/large_model_interface.yamlEnter your API Key: Find the corresponding section and paste the API Key you just copied. Here we use Tongyi Qianwen configuration as an example
xxxxxxxxxx# large_model_interface.yaml# OpenRouter平台配置 (OpenRouter Platform Configuration)openrouter_api_key"sk-xxxxxxxxxxxxxxxxxxxxxxxxxxxx"openrouter_model"nvidia/nemotron-nano-12b-v2-vl:free" # Model to use, e.g., "google/gemini-pro-vision"Open the main configuration file
yahboom.yaml:xxxxxxxxxxvim ~/yahboom_ws/src/largemodel/config/yahboom.yamlSelect the online platform to use: Modify the
llm_platformparameter to the platform name you want to usexxxxxxxxxx# yahboom.yamlmodel_serviceros__parameters# ...llm_platform'openrouter' # Currently selected large model platform# Available platforms: 'ollama','openrouter'
Recompile
xxxxxxxxxxcd ~/yahboom_ws/colcon buildsource install/setup.bash
3.2 Start and Test Function
Prepare image file:
Place a test image file in the following path:
/home/sunrise/yahboom_ws/src/largemodel/resources_file/visual_positioningThen name the image
test_image.jpgStart the
largemodelmain program:Open a terminal, then run the following command:
xxxxxxxxxxros2 launch largemodel largemodel_control.launch.pyTesting:
Wake up: Speak to the microphone: "Hello, Yahboom."
Dialogue: After the speaker responds, you can say:
Analyze the position of the dinosaur in the imageObserve logs: In the terminal running the
launchfile, you should see:- The ASR node recognizes your question and prints it out.
- The
model_servicenode receives the text, calls the LLM, and prints out the LLM's response.
Listen to answer: Shortly after, you should hear the answer from the speakers, and find an MD document recording coordinate positions and positioning object information in the path
/home/sunrise/yahboom_ws/src/largemodel/resources_file/visual_positioning.