Update yolop_demo.py
Browse files- yolop_demo.py +2 -2
yolop_demo.py
CHANGED
|
@@ -659,7 +659,7 @@ def detect(path, task, thickness, color, alpha):
|
|
| 659 |
ori_width = int(640 / shapes[1][0][1])
|
| 660 |
img_det = cv2.resize(img_det, (ori_width, ori_height), interpolation=cv2.INTER_LINEAR)
|
| 661 |
|
| 662 |
-
print('Done. (%.3fs)' % (time.time() - t0))
|
| 663 |
-
print('inf : (%.4fs/frame) nms : (%.4fs/frame)' % (inf_time.avg, nms_time.avg))
|
| 664 |
|
| 665 |
return Image.fromarray(img_det[:, :, ::-1])
|
|
|
|
| 659 |
ori_width = int(640 / shapes[1][0][1])
|
| 660 |
img_det = cv2.resize(img_det, (ori_width, ori_height), interpolation=cv2.INTER_LINEAR)
|
| 661 |
|
| 662 |
+
# print('Done. (%.3fs)' % (time.time() - t0))
|
| 663 |
+
# print('inf : (%.4fs/frame) nms : (%.4fs/frame)' % (inf_time.avg, nms_time.avg))
|
| 664 |
|
| 665 |
return Image.fromarray(img_det[:, :, ::-1])
|