Diff of /MODEL/Testing.py [000000] .. [3f73f3]

Switch to side-by-side view

--- a
+++ b/MODEL/Testing.py
@@ -0,0 +1,24 @@
+import cv2
+import numpy as np
+from ultralytics import YOLO
+
+# Load the YOLOv8 model
+model = YOLO('best_75.pt')
+
+# Load the image
+image = cv2.imread('D:\CSE\Semester - 5\Extra\IIT jammu\YOLO\Code\Test_Dataset\Test_Dataset_2\A0484.png')
+
+# Perform inference
+# results = model.predict(image, show=True, save=True)
+results = model.predict(image, show=True)
+
+# Check if the image is bleeding
+if results[0].boxes.shape[0] > 0:
+    print('Bleeding Frame')
+
+    # Print the bounded boxes
+    for box in results[0].boxes:
+        print(box)
+
+else:
+    print('Non-Bleeding Frame')
\ No newline at end of file