hank1996 commited on
Commit
59a941a
·
1 Parent(s): 1370b21

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -158,6 +158,9 @@ def detect(img,model):
158
  # Print time (inference)
159
  print(f'{s}Done. ({t2 - t1:.3f}s)')
160
  show_seg_result(im0, (da_seg_mask,ll_seg_mask), is_demo=True)
 
 
 
161
  print('Done. (%.3fs)' % (time.time() - t0))
162
  print('inf : (%.4fs/frame) nms : (%.4fs/frame)' % (inf_time.avg,nms_time.avg))
163
 
 
158
  # Print time (inference)
159
  print(f'{s}Done. ({t2 - t1:.3f}s)')
160
  show_seg_result(im0, (da_seg_mask,ll_seg_mask), is_demo=True)
161
+ inf_time.update(t2-t1,img.size(0))
162
+ nms_time.update(t4-t3,img.size(0))
163
+ #waste_time.update(tw2-tw1,img.size(0))
164
  print('Done. (%.3fs)' % (time.time() - t0))
165
  print('inf : (%.4fs/frame) nms : (%.4fs/frame)' % (inf_time.avg,nms_time.avg))
166