fix resnet50_face metric (#3310)

* fix resnet50_face metric

* fix face rec model to_static bug
pull/3324/head
学卿 2024-12-10 13:57:00 +08:00 committed by GitHub
parent 1850bca5d7
commit 070716eef1
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 9 additions and 4 deletions

View File

@ -128,4 +128,4 @@ DataLoader:
Metric:
Eval:
- BestAccOnFiveDatasets: {}
- FaceAccOnFiveDatasets: {}

View File

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