Skip to content

Commit 605ff37

Browse files
Fix cuda mp (#2023)
1 parent 3cf007a commit 605ff37

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

boxmot/engine/cli.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,7 @@ def main():
3737
)
3838
common_parser.add_argument('--imgsz', '--img-size', nargs='+', type=int,
3939
default=None, help='inference size h,w')
40-
common_parser.add_argument('--fps', type=int, default=None,
40+
common_parser.add_argument('--fps', type=int, default=30,
4141
help='video frame-rate')
4242
common_parser.add_argument('--conf', type=float, default=0.01,
4343
help='min confidence threshold')

boxmot/engine/val.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,8 @@
1-
21
# Mikel Broström 🔥 Yolo Tracking 🧾 AGPL-3.0 license
32

3+
import multiprocessing as mp
4+
mp.set_start_method("spawn", force=True)
5+
46
import argparse
57
import subprocess
68
from pathlib import Path
@@ -151,8 +153,6 @@ def generate_dets_embs(args: argparse.Namespace, y: Path, source: Path) -> None:
151153
np.savetxt(f, embs, fmt='%f')
152154

153155

154-
import re
155-
156156
def parse_mot_results(results: str) -> dict:
157157
"""
158158
Extracts COMBINED HOTA, MOTA, IDF1, AssA, AssRe, IDSW, and IDs from MOTChallenge evaluation output.

0 commit comments

Comments
 (0)