forked from mindspore-Ecosystem/mindspore
!18876 set seed for retinaface_resnet50 evaluate process
Merge pull request !18876 from zhouneng/fix_issue_I3VG2O
This commit is contained in:
commit
db71af0a6c
|
@ -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.
|
||||
|
|
Loading…
Reference in New Issue