hank1996 commited on
Commit
5db4696
·
1 Parent(s): 67d0b01

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
 
162
  if weights == 'yolop.pt':
163
  weights = 'End-to-end.pth'
 
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
+
164
 
165
  if weights == 'yolop.pt':
166
  weights = 'End-to-end.pth'