lopesdri commited on
Commit
3e1b9a6
·
1 Parent(s): 63e0b63

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -0
app.py CHANGED
@@ -25,6 +25,13 @@ def detect(img):
25
  scores = predictions[:, 4]
26
  categories = predictions[:, 5]
27
  new_image = np.squeeze(results.render())
 
 
 
 
 
 
 
28
 
29
  return new_image
30
 
 
25
  scores = predictions[:, 4]
26
  categories = predictions[:, 5]
27
  new_image = np.squeeze(results.render())
28
+ scale_percent = 60 # percent of original size
29
+ width = int(img.shape[1] * scale_percent / 100)
30
+ height = int(img.shape[0] * scale_percent / 100)
31
+ dim = (width, height)
32
+
33
+ # resize image
34
+ new_image = cv2.resize(new_image, dim, interpolation = cv2.INTER_AREA)
35
 
36
  return new_image
37