!18876 set seed for retinaface_resnet50 evaluate process

Merge pull request !18876 from zhouneng/fix_issue_I3VG2O
This commit is contained in:
i-robot 2021-06-26 02:51:07 +00:00 committed by Gitee
commit db71af0a6c
1 changed files with 4 additions and 0 deletions

View File

@ -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.