feat: Add new models

scripts/run_inference.py

@@ -4,8 +4,8 @@ import time
 import argparse
 import numpy as np
 
-from uniface.detection import RetinaFace, draw_detections
-from uniface.constants import RetinaFaceWeights
+from uniface.detection import RetinaFace, draw_detections, SCRFD
+from uniface.constants import RetinaFaceWeights, SCRFDWeights
 
 
 def run_inference(model, image_path, vis_threshold=0.6, save_dir="outputs"):

scripts/sha256_generate.py

@@ -0,0 +1,35 @@
+import argparse
+import hashlib
+from pathlib import Path
+
+
+def compute_sha256(file_path: Path, chunk_size: int = 8192) -> str:
+    sha256_hash = hashlib.sha256()
+    with file_path.open("rb") as f:
+        for chunk in iter(lambda: f.read(chunk_size), b""):
+            sha256_hash.update(chunk)
+    return sha256_hash.hexdigest()
+
+
+def main():
+    parser = argparse.ArgumentParser(
+        description="Compute SHA256 hash of a model weight file."
+    )
+    parser.add_argument(
+        "file",
+        type=Path,
+        help="Path to the model weight file (.onnx, .pth, etc)."
+    )
+
+    args = parser.parse_args()
+
+    if not args.file.exists() or not args.file.is_file():
+        print(f"File does not exist: {args.file}")
+        return
+
+    sha256 = compute_sha256(args.file)
+    print(f"`SHA256 hash for '{args.file.name}':\n{sha256}")
+
+
+if __name__ == "__main__":
+    main()