Welcome to Jetson AGX Thor repository
11.Multimodal visual understand application
Multimodal visual understand application
Multimodal visual understand application1. Concept Introduction1.1 What is "Visual Understanding"?1.2 Implementation Principle Overview2. Code AnalysisKey Code1. Tool Layer Entry (largemodel/utils/tools_manager.py)2. Model Interface Layer (largemodel/utils/large_model_interface.py)Code Analysis3.1 Configuring the Offline Large Model3.1.1 Configuring the LLM Platform (yahboom.yaml)3.1.2 Configuring the Model Interface (large_model_interface.yaml)3.2 Starting and Testing the Function (Text Input Mode)4. Common Problems and SolutionsProblem 1: The log displays "Failed to call ollama vision model" or the connection is refused.Problem 2: The seewhat tool returns "Unable to open camera" or fails to capture.
1. Concept Introduction
1.1 What is "Visual Understanding"?
In the largemodel project, the multimodal visual understanding feature enables robots to go beyond simply "seeing" a matrix of pixels and truly "understand" the content, objects, scenes, and relationships within an image. This is like giving robots a pair of thinking eyes.
The core tool for this feature is **seewhat`. When a user issues a command like "see what's here," the system invokes this tool, triggering a series of background operations that ultimately provide the user with AI-generated analysis of the live image in natural language.
1.2 Implementation Principle Overview
The basic principle is to input two different types of information—image (visual information) and text (linguistic information)—into a powerful multimodal large model (such as LLaVA).
- Image Encoding: The model first uses a vision encoder to convert the input image into computer-interpretable digital vectors. These vectors capture image features such as color, shape, and texture. 2. Text Encoding: Simultaneously, the user's question (e.g., "What's on the table?") is converted into a text vector.
- Cross-modal Fusion: In the most crucial step, the model fuses the image vector and the text vector in a special "attention layer." Here, the model learns to "focus" on the parts of the image relevant to the question. For example, when asked about "table," the model will pay more attention to areas of the image that match the characteristics of a table.
- Answer Generation: Finally, a large language model (LLM) generates a descriptive text answer based on this fused information.
Simply put, this involves highlighting the corresponding parts of the image with text, and then describing the highlighted parts with language.
2. Code Analysis
Key Code
1. Tool Layer Entry (largemodel/utils/tools_manager.py)
The seewhat function in this file defines the tool's execution flow.
x# From largemodel/utils/tools_manager.pyclass ToolsManager: # ... def seewhat(self): """ Capture camera frame and analyze environment with AI model. 捕获摄像头画面并使用AI模型分析环境。 :return: Dictionary with scene description and image path, or None if failed. """ self.node.get_logger().info("Executing seewhat() tool") image_path = self.capture_frame() if image_path: # Use isolated context for image analysis. / 使用隔离的上下文进行图像分析。 analysis_text = self._get_actual_scene_description(image_path) # Return structured data for the tool chain. / 为工具链返回结构化数据。 return { "description": analysis_text, "image_path": image_path } else: # ... (Error handling) return None def _get_actual_scene_description(self, image_path, message_context=None): """ Get AI-generated scene description for captured image. 获取捕获图像的AI生成场景描述。 :param image_path: Path to captured image file. :return: Plain text description of scene. """ try: # ... (构建Prompt) result = self.node.model_client.infer_with_image(image_path, scene_prompt, message=simple_context) # ... (处理结果) return description except Exception as e: # ...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 understanding tasks. It is responsible for calling the specific model implementation based on the configuration.
xxxxxxxxxx# From largemodel/utils/large_model_interface.pyclass model_interface: # ... def infer_with_image(self, image_path, text=None, message=None): """Unified image inference interface. / 统一的图像推理接口。""" # ... (准备消息) try: # 根据 self.llm_platform 的值,决定调用哪个具体实现 Determine 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 the Tongyi model pass # ... (其他平台的逻辑) # ... return {'response': response_content, 'messages': self.messages.copy()}Code Analysis
This feature's implementation involves two primary layers: the tool layer defines the business logic, and the model interface layer is responsible for communicating with the large language model. This layered design is key to achieving platform versatility.
Tool layer (
tools_manager.py):- The
seewhatfunction is the core of the visual understanding functionality. It encapsulates the complete process of "seeing": first, it calls thecapture_framemethod to capture the image, then calls_get_actual_scene_descriptionto prepare a prompt for the model to analyze the image. - The most critical step is calling the
infer_with_imagemethod of the model interface layer. It is not concerned with the underlying model; it is solely responsible for passing the two core data elements, "image" and "analysis instructions," to the model interface layer. - Finally, it packages the analysis results (plain text descriptions) received from the model interface layer into a structured dictionary and returns it. This allows upper-layer applications to easily use the analysis results.
- The
Model interface layer (
large_model_interface.py):- The
infer_with_imagefunction acts as a "dispatching center." Its primary responsibility is to check the current platform configuration (self.llm_platform) and, based on the configuration, dispatch tasks to specific processing functions (such asollama_inferortongyi_infer). - This layer is key to adapting to different AI platforms. All platform-specific operations (such as data encoding and API call formats) are encapsulated within their respective processing functions.
- In this way, the business logic code in
tools_manager.pyremains unchanged to support a variety of different large-scale backend model services. It simply interacts with the unified and stableinfer_with_imageinterface.
- The
In summary, the execution flow of the seewhat tool embodies a clear separation of responsibilities: ToolsManager is responsible for defining "what to do" (obtaining images and requesting analysis), while model_interface is responsible for defining "how to do it" (selecting the appropriate model platform based on the current configuration and interacting with it). This makes the tutorial's interpretation universal; the core code logic remains consistent regardless of whether the user is in online or offline mode.
3.1 Configuring the Offline Large Model
3.1.1 Configuring the LLM Platform (yahboom.yaml)
This file determines which large model platform the model_service node loads as its primary language model.
Open the file in Terminal:
xxxxxxxxxxvim ~/yahboom_ws/src/largemodel/config/yahboom.yamlModify/confirm
llm_platform:xxxxxxxxxxmodel_service#模型服务器节点参数 Model server node parametersros__parameterslanguage'en' #大模型接口语言 Large Model Interface LanguageuseolinettsFalse #文字模式下此项无效,可忽略 This item is invalid in text mode and can be ignored# 大模型配置 Large model configurationllm_platform'ollama' # 关键: 确保这里是 'ollama' Key: Make sure it's 'ollama'regional_setting "international"
3.1.2 Configuring the Model Interface (large_model_interface.yaml)
This file defines which vision model to use when the ollama platform is selected.
1.Open the file in Terminal
xxxxxxxxxxvim ~/yahboom_ws/src/largemodel/config/large_model_interface.yaml
2.Find the ollama related configuration
xxxxxxxxxx#.....## 离线大模型 (Offline Large Language Models)# Ollama配置 Ollama Configurationollama_host: "http://localhost:11434" # Ollama服务器地址 Ollama server addressollama_model: "llava" # 关键: 将这里改为你已下载的多模态模型,如 "llava" Key: Change this to the multimodal model you downloaded, such as "llava"#.....Note: Please make sure that the model specified in the configuration parameters (such as llava) can handle multimodal input.
If you have modified any configuration files, please run the following command to compile the ROS workspace for the configuration to take effect.
xxxxxxxxxxcd ~/yahboom_wscolcon build
3.2 Starting and Testing the Function (Text Input Mode)
Start the
largemodelmain program (text mode): Open a terminal and run the following command:xxxxxxxxxxros2 launch largemodel largemodel_control.launch.py text_chat_mode:=trueSend text command: Open another terminal again and run the following command,
xxxxxxxxxxros2 run text_chat text_chatThen start typing the text: "What do you see".
Observation: In the first terminal where you run the main program, you will see log output indicating that the system received the text command, invoked the
seewhattool, and ultimately printed out the text description of the desktop generated by the LLaVA model.
4. Common Problems and Solutions
Problem 1: The log displays "Failed to call ollama vision model" or the connection is refused.
Solution:
- Check the Ollama service: Run
ollama listin the terminal to ensure thatllava(or the model you configured) has been downloaded and the Ollama service is running. - Check the configuration files: Carefully verify that the configuration in
yahboom.yamlandlarge_model_interface.yamlis correct, especially the values forllm_platformandollama_model.
Problem 2: The seewhat tool returns "Unable to open camera" or fails to capture.
Solution:
- Check device connection: Run
ls /dev/video*to ensure that the system can detect your USB camera. - Permission issue: Try testing the camera using an application such as
cheeseorguvcview. If sudo works but a normal user doesn't, it may be a problem with udev rules or user group permissions.