yujia/ieemoo-ai-detecttracking
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

dettracking保存修改

Browse Source
This commit is contained in:
jiajie555
2025-04-27 16:43:12 +08:00
parent d9129c1ea3
commit 2ae9e5bb74
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
track_reid.py
View File

@ -160,7 +160,7 @@ def yolov10_resnet_tracker(
model = YOLOv10(weights) model = YOLOv10(weights)
custom = {"conf": 0.1, "batch": 1, "save": False, "mode": "predict"} custom = {"conf": 0.1, "batch": 1, "save": False, "mode": "predict"}
kwargs = {"save": True, "imgsz": 640, "conf": 0.1} kwargs = {"save": False, "imgsz": 640, "conf": 0.1}
args = {**model.overrides, **custom, **kwargs} args = {**model.overrides, **custom, **kwargs}
predictor = model.task_map[model.task]["predictor"](overrides=args, _callbacks=model.callbacks) predictor = model.task_map[model.task]["predictor"](overrides=args, _callbacks=model.callbacks)