gauthambalraj07@gmail.com commited on
Commit
d588202
·
1 Parent(s): 750f1e0

version changes in yolo

Browse files
Files changed (2) hide show
  1. requirements.txt +1 -1
  2. utils.py +4 -4
requirements.txt CHANGED
@@ -4,7 +4,7 @@ torchvision
4
  supervision==0.18.0
5
  openai==1.3.5
6
  transformers
7
- ultralytics==8.1.24
8
  azure-identity
9
  numpy
10
  opencv-python
 
4
  supervision==0.18.0
5
  openai==1.3.5
6
  transformers
7
+ ultralytics==8.2.2
8
  azure-identity
9
  numpy
10
  opencv-python
utils.py CHANGED
@@ -379,15 +379,15 @@ def predict_yolo(model, image_path, box_threshold, imgsz, scale_img, iou_thresho
379
  """
380
  # model = model['model']
381
  if scale_img:
382
- result = model.predict(
383
- source=image_path,
384
  conf=box_threshold,
385
  imgsz=imgsz,
386
  iou=iou_threshold, # default 0.7
387
  )
388
  else:
389
- result = model.predict(
390
- source=image_path,
391
  conf=box_threshold,
392
  iou=iou_threshold, # default 0.7
393
  )
 
379
  """
380
  # model = model['model']
381
  if scale_img:
382
+ result = model(
383
+ image_path,
384
  conf=box_threshold,
385
  imgsz=imgsz,
386
  iou=iou_threshold, # default 0.7
387
  )
388
  else:
389
+ result = model(
390
+ image_path,
391
  conf=box_threshold,
392
  iou=iou_threshold, # default 0.7
393
  )