From a1bb4b27c0e288c674eaeaeea491b62e635f2669 Mon Sep 17 00:00:00 2001 From: zhouneng Date: Fri, 25 Jun 2021 16:36:51 +0800 Subject: [PATCH] set seed for retinaface_resnet50 eval process --- model_zoo/official/cv/retinaface_resnet50/eval.py | 4 ++++ 1 file changed, 4 insertions(+) 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.