Spaces:
Sleeping
Sleeping
Update main.py
Browse files
main.py
CHANGED
@@ -55,7 +55,7 @@ def parse_detection(detections):
|
|
55 |
def infer(image):
|
56 |
image_arr = np.frombuffer(image, np.uint8)
|
57 |
image = cv2.imdecode(image_arr, cv2.IMREAD_COLOR)
|
58 |
-
image = cv2.resize(image, (
|
59 |
results = model(image)[0]
|
60 |
width, height = results.orig_shape[1], results.orig_shape[0]
|
61 |
print(results.speed)
|
|
|
55 |
def infer(image):
|
56 |
image_arr = np.frombuffer(image, np.uint8)
|
57 |
image = cv2.imdecode(image_arr, cv2.IMREAD_COLOR)
|
58 |
+
image = cv2.resize(image, (640, 640))
|
59 |
results = model(image)[0]
|
60 |
width, height = results.orig_shape[1], results.orig_shape[0]
|
61 |
print(results.speed)
|