Spaces:
Sleeping
Sleeping
gauthambalraj07@gmail.com
commited on
Commit
·
d588202
1
Parent(s):
750f1e0
version changes in yolo
Browse files- requirements.txt +1 -1
- 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.
|
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
|
383 |
-
|
384 |
conf=box_threshold,
|
385 |
imgsz=imgsz,
|
386 |
iou=iou_threshold, # default 0.7
|
387 |
)
|
388 |
else:
|
389 |
-
result = model
|
390 |
-
|
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 |
)
|