#2 master

Merged
jingmohan merged 5 commits from lemon/mindface_retinaface_resnet50:master into master 1 year ago
  1. +1
    -1
      eval.py
  2. +1
    -1
      infer.py

+ 1
- 1
eval.py View File

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



+ 1
- 1
infer.py View File

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


Loading…
Cancel
Save