Update iti107_app.py

iti107_app.py

@@ -37,3 +37,22 @@ model_path = "/ITI107-2024S2/9074899G/best_int8_openvino_model"
 #              inputs=gr.Image(type="pil"),
 #              outputs=gr.Image(type="pil")
 #              ).launch(share=True)
+
+# source = 'https://raw.githubusercontent.com/nyp-sit/iti107-2024S2/refs/heads/main/session-3/samples/sample_balloon.jpeg'
+source = './test_pic5.png'
+model = YOLO("iti107_assignment/train/weights/best_int8_openvino_model", task='detect')
+result = model(source, conf=0.4, iou=0.4)
+
+# Visualize the results
+for i, r in enumerate(result):
+    print(r)
+    # Plot results image
+    im_bgr = r.plot()  # BGR-order numpy array
+    im_rgb = Image.fromarray(im_bgr[..., ::-1])  # RGB-order PIL image
+
+    # Show results to screen (in supported environments)
+    r.show()
+
+    # Save results to disk
+    r.save(filename=f"results{i}.png")
+