Spaces:
Sleeping
Sleeping
jovian
commited on
Commit
·
ae429f8
1
Parent(s):
ee2d685
Add application file
Browse files
app.py
CHANGED
|
@@ -38,7 +38,8 @@ class Detection:
|
|
| 38 |
postprocess_type='NMS',
|
| 39 |
postprocess_match_metric='IOU',
|
| 40 |
postprocess_match_threshold=0.1,
|
| 41 |
-
postprocess_class_agnostic=True
|
|
|
|
| 42 |
)
|
| 43 |
|
| 44 |
# Retrieve COCO annotations
|
|
|
|
| 38 |
postprocess_type='NMS',
|
| 39 |
postprocess_match_metric='IOU',
|
| 40 |
postprocess_match_threshold=0.1,
|
| 41 |
+
postprocess_class_agnostic=True,
|
| 42 |
+
device=device
|
| 43 |
)
|
| 44 |
|
| 45 |
# Retrieve COCO annotations
|