vumichien commited on
Commit
91ba0ea
·
verified ·
1 Parent(s): cad13dc

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
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, (1920, 1920))
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)