Daniel Cerda Escobar commited on
Commit
ca59b6a
·
1 Parent(s): 2b2e54c

Visual prediction

Browse files
Files changed (1) hide show
  1. utils.py +2 -0
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"])