fix resnet50_face metric (#3310)
* fix resnet50_face metric * fix face rec model to_static bugpull/3324/head
parent
1850bca5d7
commit
070716eef1
|
@ -128,4 +128,4 @@ DataLoader:
|
|||
|
||||
Metric:
|
||||
Eval:
|
||||
- BestAccOnFiveDatasets: {}
|
||||
- FaceAccOnFiveDatasets: {}
|
||||
|
|
|
@ -66,8 +66,12 @@ def face_recognition_eval(engine, epoch_id=0):
|
|||
[images_right, paddle.flip(images_right, axis=-1)], 0)
|
||||
|
||||
with engine.auto_cast(is_eval=True):
|
||||
out_left = engine.model(images_left)
|
||||
out_right = engine.model(images_right)
|
||||
if engine.is_rec:
|
||||
out_left = engine.model(images_left, labels.reshape([-1, 1]))
|
||||
out_right = engine.model(images_right, labels.reshape([-1, 1]))
|
||||
else:
|
||||
out_left = engine.model(images_left)
|
||||
out_right = engine.model(images_right)
|
||||
|
||||
# get features
|
||||
if engine.config["Global"].get("retrieval_feature_from",
|
||||
|
@ -149,4 +153,5 @@ def face_recognition_eval(engine, epoch_id=0):
|
|||
if engine.eval_metric_func is None:
|
||||
return -1
|
||||
# return 1st metric in the dict
|
||||
return engine.eval_metric_func.avg
|
||||
return engine.eval_metric_func.avg
|
||||
|
||||
|
|
Loading…
Reference in New Issue