henry000 commited on
Commit
ce70a60
·
1 Parent(s): c289952

✏️ [Fix] typo in loading coco gt files

Browse files
Files changed (1) hide show
  1. yolo/tools/solver.py +1 -1
yolo/tools/solver.py CHANGED
@@ -224,7 +224,7 @@ class ModelValidator:
224
 
225
  with contextlib.redirect_stdout(io.StringIO()):
226
  # TODO: load with config file
227
- json_path, _ = locate_label_paths(Path(dataset_cfg.path), dataset_cfg.get("val", "val"))
228
  if json_path:
229
  self.coco_gt = COCO(json_path)
230
 
 
224
 
225
  with contextlib.redirect_stdout(io.StringIO()):
226
  # TODO: load with config file
227
+ json_path, _ = locate_label_paths(Path(dataset_cfg.path), dataset_cfg.get("validation", "val"))
228
  if json_path:
229
  self.coco_gt = COCO(json_path)
230