File size: 469 Bytes
f281b7f
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
from pathlib import Path

from huggingface_hub import snapshot_download
from xjpeg import XJPEG


snapshot = Path(snapshot_download(
    "migtissera/x-jpeg",
    allow_patterns=["model.safetensors", "config.json"],
))
codec = XJPEG(snapshot / "model.safetensors")
result = codec.compress("photo.png", output="photo.jpg", target_bpp=0.5)
print({
    "bytes": result.size_bytes,
    "bpp": result.bpp,
    "rgb_ms_ssim": result.msssim,
    "backend": result.backend,
})