From e6bd5af6a8e306a1cdef63402a77a980a04ad6e1 Mon Sep 17 00:00:00 2001
From: Grafting Rayman <156515434+GraftingRayman@users.noreply.github.com>
Date: Fri, 17 Jan 2025 11:06:44 +0000
Subject: Add files via upload

---
 r_facelib/detection/yolov5face/utils/extract_ckpt.py | 5 +++++
 1 file changed, 5 insertions(+)
 create mode 100644 r_facelib/detection/yolov5face/utils/extract_ckpt.py

(limited to 'r_facelib/detection/yolov5face/utils/extract_ckpt.py')

diff --git a/r_facelib/detection/yolov5face/utils/extract_ckpt.py b/r_facelib/detection/yolov5face/utils/extract_ckpt.py
new file mode 100644
index 0000000..e6bde00
--- /dev/null
+++ b/r_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(),'../../models/facedetection')
\ No newline at end of file
-- 
cgit v1.2.3