Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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'
|