Skip to content

Commit 4980543

Browse files
committed
fix coco_eval for not using specified mean and std
1 parent 020747f commit 4980543

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

coco_eval.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -62,7 +62,7 @@ def evaluate_coco(img_path, set_name, image_ids, coco, model, threshold=0.05):
6262
image_info = coco.loadImgs(image_id)[0]
6363
image_path = img_path + image_info['file_name']
6464

65-
ori_imgs, framed_imgs, framed_metas = preprocess(image_path, max_size=input_sizes[compound_coef])
65+
ori_imgs, framed_imgs, framed_metas = preprocess(image_path, max_size=input_sizes[compound_coef], mean=params['mean'], std=params['std'])
6666
x = torch.from_numpy(framed_imgs[0])
6767

6868
if use_cuda:

0 commit comments

Comments
 (0)