mirror of
https://github.com/yakhyo/uniface.git
synced 2025-12-30 09:02:25 +00:00
36 lines
883 B
Python
36 lines
883 B
Python
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()
|