Update app.py

app.py

@@ -252,17 +252,35 @@ def get_akc_breeds_link():
 
 
 def detect_dogs(image):
+    # 使用 YOLO 模型進行偵測
     results = model_yolo(image)
+    
+    # 打印 YOLO 偵測結果
+    print(f"YOLO detection results: {results}")
+    
     dogs = []
     for result in results:
+        # 打印每個結果
+        print(f"Result: {result}")
         for box in result.boxes:
+            # 打印邊界框的類別和座標
+            print(f"Detected class: {box.cls}, Confidence: {box.conf}, Box coordinates: {box.xyxy}")
+            
             if box.cls == 16:  # COCO 資料集中的狗類別是 16
                 xyxy = box.xyxy[0].tolist()
                 confidence = box.conf.item()
+                
+                # 確認圖片裁切過程正確
+                print(f"Cropping image at coordinates: {xyxy}")
+                
                 cropped_image = image.crop((xyxy[0], xyxy[1], xyxy[2], xyxy[3]))
                 dogs.append((cropped_image, confidence, xyxy))
+    
+    # 最後打印偵測到的狗的數量
+    print(f"Number of dogs detected: {len(dogs)}")
     return dogs
 
+
 def predict_breed(cropped_image):
     image_tensor = preprocess_image(cropped_image)
     with torch.no_grad():