Spaces:
Sleeping
Sleeping
Daniel Cerda Escobar
commited on
Commit
·
ca59b6a
1
Parent(s):
2b2e54c
Visual prediction
Browse files
utils.py
CHANGED
@@ -29,6 +29,8 @@ def sahi_yolov8m_inference(
|
|
29 |
visual_result = sahi.utils.cv.visualize_object_predictions(
|
30 |
image=numpy.array(image),
|
31 |
object_prediction_list=prediction_result.object_prediction_list,
|
|
|
|
|
32 |
)
|
33 |
|
34 |
output = Image.fromarray(visual_result["image"])
|
|
|
29 |
visual_result = sahi.utils.cv.visualize_object_predictions(
|
30 |
image=numpy.array(image),
|
31 |
object_prediction_list=prediction_result.object_prediction_list,
|
32 |
+
rect_th=2,
|
33 |
+
text_size=0.5
|
34 |
)
|
35 |
|
36 |
output = Image.fromarray(visual_result["image"])
|