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 |
)
|