glenn-jocher
commited on
Commit
•
8c5b622
1
Parent(s):
496ec33
Update detect.py
Browse files
detect.py
CHANGED
@@ -80,7 +80,7 @@ def detect(save_img=False):
|
|
80 |
p, s, im0 = path, '', im0s
|
81 |
|
82 |
save_path = str(Path(out) / Path(p).name)
|
83 |
-
txt_path =
|
84 |
s += '%gx%g ' % img.shape[2:] # print string
|
85 |
gn = torch.tensor(im0.shape)[[1, 0, 1, 0]] # normalization gain whwh
|
86 |
if det is not None and len(det):
|
|
|
80 |
p, s, im0 = path, '', im0s
|
81 |
|
82 |
save_path = str(Path(out) / Path(p).name)
|
83 |
+
txt_path = str(Path(out) / Path(p).stem) + ('_%g' % dataset.frame if dataset.mode == 'video' else '')
|
84 |
s += '%gx%g ' % img.shape[2:] # print string
|
85 |
gn = torch.tensor(im0.shape)[[1, 0, 1, 0]] # normalization gain whwh
|
86 |
if det is not None and len(det):
|