Puyush commited on
Commit
1c9bdd1
1 Parent(s): 043df01

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -88,8 +88,8 @@ def image_predict(image):
88
  res = check_acc(box)
89
  annotated_frame = results[0].plot()
90
  if len(res) >0:
91
- annotated_frame_bgr = cv2.cvtColor(annotated_frame, cv2.COLOR_RGB2BGR)
92
- send_email(res, annotated_frame_bgr)
93
  return (res, annotated_frame)
94
 
95
  return ("No Class Detected", None)
@@ -118,8 +118,8 @@ def extract_frames(video):
118
  annotated_frame = results[0].plot()
119
 
120
  if len(res) >0:
121
- annotated_frame_bgr = cv2.cvtColor(annotated_frame, cv2.COLOR_RGB2BGR)
122
- send_email(res, annotated_frame_bgr)
123
  return (res, annotated_frame)
124
 
125
  frame_no += 1 # Increment frame number
 
88
  res = check_acc(box)
89
  annotated_frame = results[0].plot()
90
  if len(res) >0:
91
+ # annotated_frame_bgr = cv2.cvtColor(annotated_frame, cv2.COLOR_RGB2BGR)
92
+ # send_email(res, annotated_frame_bgr)
93
  return (res, annotated_frame)
94
 
95
  return ("No Class Detected", None)
 
118
  annotated_frame = results[0].plot()
119
 
120
  if len(res) >0:
121
+ # annotated_frame_bgr = cv2.cvtColor(annotated_frame, cv2.COLOR_RGB2BGR)
122
+ # send_email(res, annotated_frame_bgr)
123
  return (res, annotated_frame)
124
 
125
  frame_no += 1 # Increment frame number