Procházet zdrojové kódy

detect.py fast bug fix

5.0
Glenn Jocher před 4 roky
rodič
revize
bbd12c7cfe
1 změnil soubory, kde provedl 1 přidání a 2 odebrání
  1. +1
    -2
      detect.py

+ 1
- 2
detect.py Zobrazit soubor

@@ -64,8 +64,7 @@ def detect(save_img=False):
pred = model(img, augment=opt.augment)[0]

# Apply NMS
pred = non_max_suppression(pred, opt.conf_thres, opt.iou_thres,
fast=True, classes=opt.classes, agnostic=opt.agnostic_nms)
pred = non_max_suppression(pred, opt.conf_thres, opt.iou_thres, classes=opt.classes, agnostic=opt.agnostic_nms)
t2 = torch_utils.time_synchronized()

# Apply Classifier

Načítá se…
Zrušit
Uložit