[update] update on the image test set

IoU_yolo.py

@@ -2,11 +2,9 @@ import os
 import sys 
 import numpy as np
 
+# python IoU_yolo.py test_images/labels test/model/labels
 
-# GT_DIR = "test_images/wit396_pdf545/labels"
 GT_DIR = sys.argv[1]
-
-# PRED_DIR = "test/yolo8Lg-17-07-2026/labels"
 PRED_DIR = sys.argv[2]
  
 def yolo_to_corners(box):

inference.py

@@ -3,7 +3,7 @@ import glob
 import os
 from ultralytics import YOLO
 
-# uv run predict.py path_to_model.pt path_to_images ./test
+# uv run inference.py path_to_model.pt path_to_images ./test
 
 
 def parse_args() -> argparse.Namespace:

test_images/labels/1.txt

@@ -0,0 +1,4 @@
+0 0.542379 0.785400 0.149692 0.161200
+0 0.107314 0.773200 0.202324 0.172000
+0 0.139781 0.587000 0.153794 0.166000
+0 0.569378 0.590800 0.120301 0.160800

test_images/labels/2.txt

@@ -0,0 +1,8 @@
+0 0.735294 0.299800 0.221614 0.149200
+0 0.444596 0.615200 0.235294 0.130400
+0 0.449384 0.464400 0.229822 0.140000
+0 0.745212 0.463600 0.211354 0.141600
+0 0.738030 0.786000 0.207934 0.144000
+0 0.454514 0.306800 0.225034 0.151200
+0 0.452804 0.787600 0.221614 0.143200
+0 0.725718 0.618800 0.242134 0.136000

test_images/labels/3.txt

@@ -0,0 +1,8 @@
+0 0.199109 0.458000 0.236463 0.151200
+0 0.180603 0.289600 0.209047 0.146400
+0 0.193968 0.621600 0.154901 0.153600
+0 0.465730 0.460000 0.236463 0.145600
+0 0.468129 0.618000 0.168609 0.146400
+0 0.485949 0.798400 0.126114 0.167200
+0 0.209390 0.780000 0.147361 0.150400
+0 0.459561 0.296600 0.217272 0.154800

test_images/labels/6.txt

@@ -0,0 +1,20 @@
+0 0.197360 0.261400 0.172219 0.118000
+0 0.186989 0.539800 0.169076 0.121200
+0 0.373350 0.673200 0.144563 0.130400
+0 0.544312 0.535800 0.164676 0.124400
+0 0.178818 0.674400 0.180390 0.129600
+0 0.387178 0.815000 0.096794 0.131600
+0 0.189818 0.399400 0.158391 0.112400
+0 0.726587 0.533800 0.150849 0.126800
+0 0.207102 0.812800 0.106223 0.130400
+0 0.546826 0.253800 0.160905 0.129200
+0 0.734758 0.262200 0.152106 0.118000
+0 0.728473 0.672800 0.173476 0.128000
+0 0.739158 0.811600 0.129478 0.124800
+0 0.376493 0.257800 0.170962 0.124400
+0 0.545883 0.398200 0.155248 0.125200
+0 0.368322 0.533800 0.163419 0.130800
+0 0.540541 0.671400 0.174733 0.134000
+0 0.368322 0.400000 0.152106 0.118400
+0 0.555625 0.815400 0.130735 0.124400
+0 0.720302 0.399000 0.162162 0.121200

test_images/labels/7.txt

@@ -0,0 +1,18 @@
+0 0.225312 0.570417 0.105625 0.124167
+0 0.230937 0.402917 0.119375 0.169167
+0 0.376563 0.387917 0.108125 0.184167
+0 0.387813 0.877083 0.080625 0.147500
+0 0.237813 0.888750 0.075625 0.140833
+0 0.731875 0.826250 0.106250 0.172500
+0 0.627812 0.823333 0.098125 0.198333
+0 0.375937 0.230000 0.134375 0.188333
+0 0.540937 0.652917 0.098125 0.197500
+0 0.638125 0.442917 0.107500 0.197500
+0 0.549687 0.447917 0.076875 0.205833
+0 0.237813 0.235000 0.114375 0.173333
+0 0.742812 0.448750 0.079375 0.192500
+0 0.383750 0.559167 0.095000 0.150000
+0 0.730000 0.653333 0.105000 0.166667
+0 0.386562 0.722500 0.089375 0.161667
+0 0.226875 0.734167 0.111250 0.170000
+0 0.633750 0.645833 0.088750 0.183333