浏览代码

update times to end-to-end latency (inference + NMS)

5.0
Glenn Jocher 4 年前
父节点
当前提交
ad71d2d513
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. +1
    -1
      detect.py

+ 1
- 1
detect.py 查看文件

@@ -62,7 +62,6 @@ def detect(save_img=False):
# Inference
t1 = torch_utils.time_synchronized()
pred = model(img, augment=opt.augment)[0]
t2 = torch_utils.time_synchronized()

# to float
if half:
@@ -71,6 +70,7 @@ def detect(save_img=False):
# Apply NMS
pred = non_max_suppression(pred, opt.conf_thres, opt.iou_thres,
fast=True, classes=opt.classes, agnostic=opt.agnostic_nms)
t2 = torch_utils.time_synchronized()

# Apply Classifier
if classify:

正在加载...
取消
保存