CodeFormer
CodeFormer copied to clipboard
Modify the effect and face detection strength
trafficstars
Hi, Thank you for amazing project.
How to modify the effect and face detection strength?
We have provided four different models for face detection, you can set --detection_model as one of retinaface_resnet50, retinaface_mobile0.25, YOLOv5l, and YOLOv5n to compare these models and choose the best one for your testing.
FYI:
parser.add_argument('--detection_model', type=str, default='retinaface_resnet50',
help='Face detector. Optional: retinaface_resnet50, retinaface_mobile0.25, YOLOv5l, YOLOv5n.
Default: retinaface_resnet50')