Add codeformer and update license

2025-04-09 23:03:26 +01:00
parent f40ffde905
commit 3549b7e50c
84 changed files with 12545 additions and 19 deletions
--- a/facelib/detection/yolov5face/utils/extract_ckpt.py
+++ b/facelib/detection/yolov5face/utils/extract_ckpt.py
@@ -0,0 +1,5 @@
+import torch
+import sys
+sys.path.insert(0,'./facelib/detection/yolov5face')
+model = torch.load('facelib/detection/yolov5face/yolov5n-face.pt', map_location='cpu')['model']
+torch.save(model.state_dict(),'weights/facelib/yolov5n-face.pth')