diff --git a/model_zoo/official/cv/retinaface_resnet50/eval.py b/model_zoo/official/cv/retinaface_resnet50/eval.py index cc789c9820f..08edb70c0d7 100644 --- a/model_zoo/official/cv/retinaface_resnet50/eval.py +++ b/model_zoo/official/cv/retinaface_resnet50/eval.py @@ -21,12 +21,16 @@ import numpy as np import cv2 from mindspore import Tensor, context +from mindspore.common import set_seed from mindspore.train.serialization import load_checkpoint, load_param_into_net from src.config import cfg_res50 from src.network import RetinaFace, resnet50 from src.utils import decode_bbox, prior_box + +set_seed(1) + class Timer(): def __init__(self): self.start_time = 0.