diff --git a/eval.py b/eval.py index d9ba1cc..abe2b66 100644 --- a/eval.py +++ b/eval.py @@ -215,7 +215,7 @@ if __name__ == '__main__': help='device where the code will be implemented (default: Ascend),if to use the CPU on the Qizhi platform:device_target=CPU') - parser.add_argument('--config', default='RetinaFace_mobilenet025.yaml', type=str ,help='config path') + parser.add_argument('--config', default='RetinaFace_resnet50.yaml', type=str ,help='config path') args = parser.parse_args() diff --git a/infer.py b/infer.py index cb7e30e..1feab6a 100644 --- a/infer.py +++ b/infer.py @@ -130,7 +130,7 @@ def infer(cfg): if __name__ == '__main__': parser = argparse.ArgumentParser(description='infer') - parser.add_argument('--config', default='mindface/detection/configs/RetinaFace_mobilenet025.yaml', type=str, + parser.add_argument('--config', default='mindface/detection/configs/RetinaFace_resnet50.yaml', type=str, help='configs path') parser.add_argument('--checkpoint', type=str, default='', help='checpoint path')