OAOA commited on
Commit
ea3f9a7
1 Parent(s): a5f20a6

change the detection model to retinaface

Browse files
configs/sample/iddpm_ffhq512_swinir.yaml CHANGED
@@ -60,5 +60,5 @@ model_ir:
60
 
61
  # face detection model for unaligned face
62
  detection:
63
- det_model: "YOLOv5l" # large model: 'YOLOv5l', 'retinaface_resnet50'; small model: 'YOLOv5n', 'retinaface_mobile0.25'
64
  upscale: 2 # The final upscaling factor for the whole image
60
 
61
  # face detection model for unaligned face
62
  detection:
63
+ det_model: "retinaface_resnet50" # large model: 'YOLOv5l', 'retinaface_resnet50'; small model: 'YOLOv5n', 'retinaface_mobile0.25'
64
  upscale: 2 # The final upscaling factor for the whole image