fix: continue when object detected

main.py

@@ -51,7 +51,9 @@ def animal_detect_and_classify(img_path):
     # Iterate over the detected objects
     # Iterate over detections
     for result in detect_results:
+        flag = False
         for box in result.boxes:
+            flag = True
             # print(box)
             # Crop the RoI
             x1, y1, x2, y2 = map(int, box.xyxy[0])
@@ -78,6 +80,8 @@ def animal_detect_and_classify(img_path):
             predicted_labels = [labels[np.argmax(pred)] if np.max(pred) >= threshold else "animal" for pred in prediction]
             print(predicted_labels)
             combined_results.append(((x1, y1, x2, y2), predicted_labels))
+        if flag:
+            continue
         y2, x2, _ = img.shape
         detect_img = img[0:y2, 0:x2]
         detect_img = cv2.cvtColor(detect_img, cv2.COLOR_BGR2RGB)