Feature visualization improvements 32 (#3947)

This commit is contained in:
Glenn Jocher authored and GitHub committed 2021-07-09 15:23:02 +02:00
1 parent dabad5793a
commit 248504cf13
2 files changed
+7 -6

No files matched your search

+1 -1
View File
@@ -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)