Spaces:
Runtime error
Runtime error
Remove GPU in module
Browse files- detect_strongsort.py +0 -2
detect_strongsort.py
CHANGED
|
@@ -1,4 +1,3 @@
|
|
| 1 |
-
import spaces
|
| 2 |
import argparse
|
| 3 |
|
| 4 |
import os
|
|
@@ -56,7 +55,6 @@ def plot_one_box(x, img, color=None, label=None, line_thickness=3):
|
|
| 56 |
cv2.putText(img, label, (c1[0], c1[1] - 2), 0, tl / 3, [225, 255, 255], thickness=tf, lineType=cv2.LINE_AA)
|
| 57 |
|
| 58 |
|
| 59 |
-
@spaces.GPU(duration=120)
|
| 60 |
@smart_inference_mode()
|
| 61 |
def run(
|
| 62 |
source='0',
|
|
|
|
|
|
|
| 1 |
import argparse
|
| 2 |
|
| 3 |
import os
|
|
|
|
| 55 |
cv2.putText(img, label, (c1[0], c1[1] - 2), 0, tl / 3, [225, 255, 255], thickness=tf, lineType=cv2.LINE_AA)
|
| 56 |
|
| 57 |
|
|
|
|
| 58 |
@smart_inference_mode()
|
| 59 |
def run(
|
| 60 |
source='0',
|