Welcome to RDK S100(P) Board repository
7. Multimodal visual localization application
Multimodal visual localization application
Multimodal visual localization application1. Concept Introduction1.1 What is "Multimodal Visual Localization"?1.2 Brief Description of Implementation Principles2. Code AnalysisKey Code1. Tool Layer Entry Point (largemodel/utils/tools_manager.py)2. Model Interface Layer (largemodel/utils/large_model_interface.py)Code Analysis3. Practical Operation3.1 Configuring Offline Large Models3.1.1 Configuring the LLM Platform (yahboom.yaml)3.1.2 Configure the model interface (large_model_interface.yaml)3.1.3 Recompile3.2 Launching and Testing the Feature
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 the location and posture of devices or users in their environment. This technology does not rely solely on a single type of sensor data but integrates information from different perceptual modalities, thereby improving the accuracy and robustness of localization.
1.2 Brief Description of Implementation Principles
- 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 image-processing-appropriate architectures to extract features and map them into the same embedding space as the text.
- Joint Training: By designing an appropriate loss function, text and visual data can be trained simultaneously within the same framework, allowing the model to learn to establish connections between the two modalities. For example, in a question-answering system, answers can be provided based on both text questions and related image content.
- Visually Guided Language Generation/Understanding: Once an effective cross-modal representation is established, visual information can be used to enhance the functionality of the language model. For instance, given a photograph, the model can not only describe what is 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 location).
2. Code Analysis
Key Code
1. Tool Layer Entry Point (largemodel/utils/tools_manager.py)
The visual_positioning function in this file defines the execution flow of the tool, specifically how it constructs a Prompt containing the target object's name and formatting 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 check) # Construct a prompt asking the large model to identify the coordinates of the specified object. if self.node.language == 'zh': prompt = f"请仔细分析这张图片,用一个个框定位图像每一个{object_name}的位置..." else: prompt = f"Please carefully analyze this image and find the position of all {object_name}..." # ... (Building an independent message context) result = self.node.model_client.infer_with_image(image_path, prompt, message=message_to_use) # ... (Processing and parsing the 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. """ # ... (Preparing the message) try: # The value of self.llm_platform determines which specific implementation to call. if self.llm_platform == 'ollama': response_content = self.ollama_infer(self.messages, image_path=image_path) elif self.llm_platform == 'tongyi': # ... Logic of calling the generalized model pass # ... (Logic for other platforms) # ... return {'response': response_content, 'messages': self.messages.copy()}Code Analysis
The core of the visual positioning function lies in guiding a large model to output structured data through precise instructions. It also follows a layered design of tool layer and model interface layer.
- Tool Layer (
tools_manager.py):
- The
visual_positioningfunction is the core of this function's business logic. It receives two key parameters:image_path(image path) andobject_name(the name of the object to be positioned). - The most crucial operation of this function is constructing a highly customized Prompt. It doesn't simply request the model to describe the image; instead, it embeds
object_nameinto a carefully designed template, explicitly instructing the model to "position each {object_name} in the image," and implicitly or explicitly requesting the return of results in a specific format (such as a coordinate array). - After constructing the Prompt, it calls the
infer_with_imagemethod of the model interface layer, passing the image and this customized instruction along with it. - After receiving the returned text from the model interface layer, it still needs to perform post-processing: using methods such as 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-layer application.
- Model Interface Layer (
large_model_interface.py):
- The
infer_with_imagefunction still plays the role of the "dispatch center." It receives the image and prompt fromvisual_positioningand distributes the task to the correct backend model implementation based on the current configuration (self.llm_platform). - For visual positioning tasks, the model interface layer's responsibilities are essentially the same as for visual understanding tasks: correctly package the image data and text instructions, send them to the selected model platform, and then return the text results intact to the tool layer. All platform-specific implementation details are encapsulated in this layer.
In summary, the general process of visual localization is as follows: ToolsManager receives the name of the target object and constructs a precise Prompt that requests coordinates -> ToolsManager calls the model interface -> model_interface packages the image and the Prompt together and sends it to the appropriate model platform according to the configuration -> The model returns text containing coordinate information -> model_interface returns the text to ToolsManager -> ToolsManager parses the text, extracts the structured coordinate data, and returns it. This process demonstrates how Prompt Engineering technology allows general-purpose large-scale visual models to perform more specific and structured tasks.
3. Practical Operation
3.1 Configuring Offline Large Models
3.1.1 Configuring the LLM Platform (yahboom.yaml)
This file determines which large-scale model platform the model_service node loads as its primary language model.
- Open the file in the terminal:
xxxxxxxxxxvim ~/yahboom_ws/src/largemodel/config/yahboom.yaml- Modify/confirm
llm_platform:
xxxxxxxxxxmodel_service# Model server node parameters ros__parameters language'en' # Large model interface language useolinettsTrue # This option is invalid in text mode and can be ignored # Large model configuration llm_platform'ollama' # Critical: Ensure this is 'ollama'3.1.2 Configure the model interface (large_model_interface.yaml)
This file defines which visual model is used when the platform is selected as ollama.
- Open the file in the terminal
xxxxxxxxxxvim ~/yahboom_ws/src/largemodel/config/large_model_interface.yaml
- Locate the Ollama-related configuration
xxxxxxxxxx#.....## Offline Large Language Models# Ollama Configurationollama_host: "http://localhost:11434" # Ollama server addressollama_model: "llava" # Key: Replace this with your downloaded multimodal model, such as "llava"#.....Note: Ensure that the model specified in the configuration parameters (such as llava) can handle multimodal input.
3.1.3 Recompile
xxxxxxxxxxcd ~/yahboom_ws/colcon buildsource install/setup.bash
3.2 Launching and Testing the Feature
- Preparing the Image File:
Place the image file to be tested in the following path: /home/sunrise/yahboom_ws/src/largemodel/resources_file/visual_positioning
Then name the image test_image.jpg
- Launching the
largemodelMain Program:
Open a terminal and run the following command:
xxxxxxxxxxros2 launch largemodel largemodel_control.launch.py- Testing:
- Wake-up: Say into the microphone, "Hello, yahboom."
- Dialogue: After the speaker responds, you can say:
Analyze the position of the dinosaur in the image - Observing the Log: 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 the LLM's response.
- Listen to the response: Shortly after, you should hear the response from the speaker and find a Markdown file containing coordinates and object positioning information in the path
/home/sunrise/yahboom_ws/src/largemodel/resources_file/visual_positioning.