Upload app.py

app.py

@@ -719,8 +719,22 @@ def run_model(
     # Load segmentation models if enabled (使用CPU节省GPU资源)
     if enable_segmentation:
         progress(0.1, desc="🎯 加载分割模型 (CPU)...")
+        print(f"\n{'='*70}")
+        print("🎯 分割模型加载开始...")
+        print(f"{'='*70}")
         load_grounding_dino_model("cpu")  # 分割使用CPU
         load_sam_model("cpu")  # MobileSAM在CPU上运行良好
+        
+        # 验证模型是否成功加载
+        if grounding_dino_model is None:
+            print("❌ GroundingDINO 模型加载失败！")
+            return predictions, processed_data, None
+        if sam_predictor is None:
+            print("❌ SAM 模型加载失败！")
+            return predictions, processed_data, None
+        
+        print(f"✅ 所有分割模型加载成功")
+        print(f"{'='*70}\n")
 
     # Load images
     progress(0.15, desc="📷 加载图片...")
@@ -795,8 +809,12 @@ def run_model(
     segmented_glb = None
     if enable_segmentation and grounding_dino_model is not None:
         progress(0.65, desc="🎯 开始物体分割...")
-        print("\n🎯 Starting segmentation...")
-        print(f"🔍 Detection prompt: {text_prompt[:100]}...")
+        print(f"\n{'='*70}")
+        print("🎯 开始物体分割...")
+        print(f"🔍 检测提示词: {text_prompt[:100]}...")
+        print(f"📊 置信度阈值: {GROUNDING_DINO_BOX_THRESHOLD}")
+        print(f"📐 最小掩码面积: {MIN_MASK_AREA} px")
+        print(f"{'='*70}\n")
 
         all_view_detections = []
         all_view_masks = []
@@ -812,8 +830,11 @@ def run_model(
                 ref_image_np = ref_image
 
             detections = run_grounding_dino_detection(ref_image_np, text_prompt, "cpu")  # 使用CPU进行检测
-
+            print(f"   ✓ 检测到 {len(detections)} 个物体")
+            
             if len(detections) > 0:
+                for i, det in enumerate(detections):
+                    print(f"      物体 {i+1}: {det['label']} (置信度: {det['confidence']:.2f})")
                 boxes = [d['bbox'] for d in detections]
                 masks = run_sam_refinement(ref_image_np, boxes)
 
@@ -850,7 +871,14 @@ def run_model(
             else:
                 print(f"⚠️ 分割3D模型生成失败")
         else:
+            print(f"\n{'='*70}")
             print(f"⚠️ 未检测到任何物体，无法生成分割模型")
+            print(f"\n💡 调试提示:")
+            print(f"   1. 检查检测提示词是否准确（当前: {text_prompt[:50]}...）")
+            print(f"   2. 当前置信度阈值: {GROUNDING_DINO_BOX_THRESHOLD}")
+            print(f"   3. 尝试更通用的提示词，如: {COMMON_OBJECTS_PROMPT[:80]}...")
+            print(f"   4. 确保图片中有清晰可见的物体")
+            print(f"{'='*70}\n")
 
     # Cleanup
     progress(0.95, desc="🧹 清理内存...")
@@ -1685,6 +1713,26 @@ CUSTOM_CSS = GRADIO_CSS + """
 .file-upload-enhanced {
     position: relative;
 }
+
+/* 减少Accordion和后续内容的间距 */
+.accordion {
+    margin-bottom: 10px !important;
+}
+
+/* 页脚样式 */
+.footer {
+    margin-top: 20px !important;
+    padding: 10px 0 !important;
+}
+
+/* 信息提示框 */
+.info-box {
+    background-color: #E3F2FD !important;
+    border-left: 4px solid #2196F3 !important;
+    padding: 15px !important;
+    margin-bottom: 15px !important;
+    border-radius: 4px !important;
+}
 """
 
 # JavaScript for paste support
@@ -1799,8 +1847,19 @@ with gr.Blocks(theme=theme, css=CUSTOM_CSS, title="MapAnything V2 - 3D重建与
                     )
                 
                 with gr.Tab("🎨 分割3D"):
+                    gr.Markdown(
+                        """
+                        💡 **使用说明**:
+                        1. 在下方「⚙️ 高级选项」中勾选「启用语义分割 (CPU)」
+                        2. 点击「开始重建」按钮
+                        3. 等待处理完成后，分割结果将显示在此处
+                        
+                        📌 如果没有显示分割结果，请查看控制台日志查找原因
+                        """,
+                        elem_classes=["info-box"]
+                    )
                     segmented_output = gr.Model3D(
-                        height=550, zoom_speed=0.5, pan_speed=0.5,
+                        height=450, zoom_speed=0.5, pan_speed=0.5,
                         clear_color=[0.0, 0.0, 0.0, 0.0]
                     )
                 
@@ -1923,6 +1982,18 @@ with gr.Blocks(theme=theme, css=CUSTOM_CSS, title="MapAnything V2 - 3D重建与
                                         target_dir_output, image_gallery, log_output
                                     ]
                                 )
+    
+    # 页脚
+    gr.Markdown("---")  # 分隔线
+    gr.Markdown(
+        """
+        <div style="text-align: center; padding: 20px 0; color: #666;">
+            <p>MapAnything V2 - 3D重建与物体分割系统</p>
+            <p style="font-size: 14px;">基于 DBSCAN 聚类的多视图融合 | CPU优化的轻量级分割</p>
+        </div>
+        """,
+        elem_classes=["footer"]
+    )
 
     # === 事件绑定 ===