Spaces:
Sleeping
Sleeping
app final worked
Browse files
__pycache__/xyxy_converter.cpython-310.pyc
ADDED
|
Binary file (1.32 kB). View file
|
|
|
app.py
CHANGED
|
@@ -48,19 +48,19 @@ def main():
|
|
| 48 |
|
| 49 |
os.chdir(YOLO_PATH)
|
| 50 |
|
| 51 |
-
try:
|
| 52 |
-
|
| 53 |
-
except:
|
| 54 |
-
|
| 55 |
|
| 56 |
-
image_path =
|
| 57 |
# command = f"python detect.py --weights {MODEL_PATH} --source {image_path} --img 640 --conf 0.4 --name 'temp_exp' --hide-labels --hide-conf --save-txt --exist-ok"
|
| 58 |
command = f'''
|
| 59 |
python detect.py \
|
| 60 |
--weights {MODEL_PATH} \
|
| 61 |
--source {image_path} \
|
| 62 |
--img 640 \
|
| 63 |
-
--conf 0.
|
| 64 |
--name temp_exp \
|
| 65 |
--hide-labels \
|
| 66 |
--hide-conf \
|
|
@@ -106,7 +106,8 @@ def main():
|
|
| 106 |
except:
|
| 107 |
pass
|
| 108 |
else:
|
| 109 |
-
|
|
|
|
| 110 |
|
| 111 |
# Resize or preprocess the image as needed for your model
|
| 112 |
# For example, resizing to a specific input size
|
|
|
|
| 48 |
|
| 49 |
os.chdir(YOLO_PATH)
|
| 50 |
|
| 51 |
+
# try:
|
| 52 |
+
# shutil.rmtree('runs/detect/temp_exp')
|
| 53 |
+
# except:
|
| 54 |
+
# pass
|
| 55 |
|
| 56 |
+
image_path = "../odometer_image.jpg"
|
| 57 |
# command = f"python detect.py --weights {MODEL_PATH} --source {image_path} --img 640 --conf 0.4 --name 'temp_exp' --hide-labels --hide-conf --save-txt --exist-ok"
|
| 58 |
command = f'''
|
| 59 |
python detect.py \
|
| 60 |
--weights {MODEL_PATH} \
|
| 61 |
--source {image_path} \
|
| 62 |
--img 640 \
|
| 63 |
+
--conf 0.5 \
|
| 64 |
--name temp_exp \
|
| 65 |
--hide-labels \
|
| 66 |
--hide-conf \
|
|
|
|
| 106 |
except:
|
| 107 |
pass
|
| 108 |
else:
|
| 109 |
+
display_text = "An odometer is not detected in the image!!!"
|
| 110 |
+
processed_image = cv2.imread('odometer_image.jpg')
|
| 111 |
|
| 112 |
# Resize or preprocess the image as needed for your model
|
| 113 |
# For example, resizing to a specific input size
|
odometer_image.jpg
ADDED
|
xyxy_converter.py
CHANGED
|
@@ -13,8 +13,10 @@ def yolov5_to_image_coordinates(text_file, image_width=640, image_height=640):
|
|
| 13 |
|
| 14 |
with open(text_file, 'r') as f:
|
| 15 |
xywh_text = f.read()
|
| 16 |
-
|
| 17 |
-
|
|
|
|
|
|
|
| 18 |
|
| 19 |
x_center, y_center, width, height = yolo_box
|
| 20 |
|
|
|
|
| 13 |
|
| 14 |
with open(text_file, 'r') as f:
|
| 15 |
xywh_text = f.read()
|
| 16 |
+
|
| 17 |
+
xywh_splitted = xywh_text.split('\n')[0]
|
| 18 |
+
yolo_box = [float(i) for i in xywh_splitted.split(' ')[1:]]
|
| 19 |
+
# yolo_box = [float(i) for i in xywh_text.replace('\n', '').split(' ')[1:]]
|
| 20 |
|
| 21 |
x_center, y_center, width, height = yolo_box
|
| 22 |
|
yolov5/__pycache__/export.cpython-310.pyc
ADDED
|
Binary file (31.6 kB). View file
|
|
|
yolov5/runs/detect/temp_exp/labels/odometer_image.txt
ADDED
|
@@ -0,0 +1,10 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
0 0.635938 0.4625 0.30625 0.325
|
| 2 |
+
0 0.452344 0.715625 0.429688 0.18125
|
| 3 |
+
0 0.452344 0.715625 0.429688 0.18125
|
| 4 |
+
0 0.409375 0.657031 0.375 0.148438
|
| 5 |
+
0 0.409375 0.657031 0.375 0.148438
|
| 6 |
+
0 0.409375 0.657031 0.375 0.148438
|
| 7 |
+
0 0.564844 0.671094 0.207813 0.0859375
|
| 8 |
+
0 0.442187 0.621094 0.38125 0.139062
|
| 9 |
+
0 0.510156 0.665625 0.254687 0.11875
|
| 10 |
+
0 0.539062 0.586719 0.375 0.201562
|
yolov5/runs/detect/temp_exp/odometer_image.jpg
ADDED
|