Update mtpnet_demo.py
Browse files- mtpnet_demo.py +2 -2
mtpnet_demo.py
CHANGED
|
@@ -95,8 +95,8 @@ def detect(path, task, thickness, color, alpha):
|
|
| 95 |
ori_width = int(640 / shapes[1][0][1])
|
| 96 |
img_det = cv2.resize(img_det, (ori_width, ori_height), interpolation=cv2.INTER_LINEAR)
|
| 97 |
|
| 98 |
-
print('Done. (%.3fs)' % (time.time() - t0))
|
| 99 |
-
print('inf : (%.4fs/frame) nms : (%.4fs/frame)' % (inf_time.avg, nms_time.avg))
|
| 100 |
|
| 101 |
return Image.fromarray(img_det[:, :, ::-1])
|
| 102 |
|
|
|
|
| 95 |
ori_width = int(640 / shapes[1][0][1])
|
| 96 |
img_det = cv2.resize(img_det, (ori_width, ori_height), interpolation=cv2.INTER_LINEAR)
|
| 97 |
|
| 98 |
+
# print('Done. (%.3fs)' % (time.time() - t0))
|
| 99 |
+
# print('inf : (%.4fs/frame) nms : (%.4fs/frame)' % (inf_time.avg, nms_time.avg))
|
| 100 |
|
| 101 |
return Image.fromarray(img_det[:, :, ::-1])
|
| 102 |
|