Feature visualization improvements 32 (#3947)
This commit is contained in:
1 parent
dabad5793a
commit
248504cf13
2 files changed
+7
-6
No files matched your search
@@ -103,7 +103,7 @@ def run(weights='yolov5s.pt', # model.pt path(s)
|
||||
t1 = time_synchronized()
|
||||
pred = model(img,
|
||||
augment=augment,
|
||||
visualize=increment_path(save_dir / 'features', mkdir=True) if visualize else False)[0]
|
||||
visualize=increment_path(save_dir / Path(path).stem, mkdir=True) if visualize else False)[0]
|
||||
|
||||
# Apply NMS
|
||||
pred = non_max_suppression(pred, conf_thres, iou_thres, classes, agnostic_nms, max_det=max_det)
|
||||
|
||||
Reference in new issue
Block a user