Skip to content

Commit

Permalink
Merge pull request #1795 from mikel-brostrom/fix-loader
Browse files Browse the repository at this point in the history
ultra dataloader
  • Loading branch information
mikel-brostrom authored Jan 18, 2025
2 parents 102ed4a + cae3a02 commit 7f5dcea
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions tracking/val.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,10 +21,10 @@
from boxmot.utils import ROOT, WEIGHTS, TRACKER_CONFIGS, logger as LOGGER, EXAMPLES, DATA
from boxmot.utils.checks import RequirementsChecker
from boxmot.utils.torch_utils import select_device
from boxmot.data.loader import LoadImagesAndVideos
from boxmot.utils.misc import increment_path

from ultralytics import YOLO
from ultralytics.data.loaders import LoadImagesAndVideos

from tracking.detectors import (get_yolo_inferer, default_imgsz,
is_ultralytics_model, is_yolox_model)
Expand Down Expand Up @@ -300,7 +300,7 @@ def parse_mot_results(results: str) -> dict:
dict: A dictionary containing HOTA, MOTA, and IDF1 scores.
"""
combined_results = results.split('COMBINED')[2:-1]
combined_results = [float(re.findall(r"[-+]?(?:\d*\.*\d+)", f)[0])
combined_results = [float(re.findall("[-+]?(?:\d*\.*\d+)", f)[0])
for f in combined_results]

results_dict = {}
Expand Down

0 comments on commit 7f5dcea

Please sign in to comment.