coco128 val datasets

2e64a0f4 · Rayyyyy · 1c8fb024 · 2e64a0f4
Commit 2e64a0f4 authored May 17, 2024 by Rayyyyy
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

test.py test.py +1 -1

No files found.
--- a/test.py
+++ b/test.py
@@ -44,7 +44,7 @@ def test_img(args, model, postprocessors, save_path):
        min_score = 0.65
        img_name = dataset_test.coco.loadImgs(targets[0]['image_id'].item())[0]['file_name']
-        img = cv2.imread(os.path.join(args.coco_path, 'images/val2017', img_name))
+        img = cv2.imread(os.path.join(args.coco_path, 'val2017', img_name))
        draw_img = img.copy()
        save_status = False