wip: C++ 参考实现(待编译验证)
This commit is contained in:
@@ -0,0 +1,31 @@
|
||||
#!/usr/bin/env python
|
||||
"""把图片转成连续的 HWC uint8 缓冲,供 C++ 参考实现读取与对拍。
|
||||
|
||||
uv run python scripts/dump_raw.py 某张图.png /tmp/image.bin
|
||||
|
||||
打印 `宽 高 通道`,直接喂给 deploy/cpp 的 film_predict。
|
||||
"""
|
||||
|
||||
import sys
|
||||
from pathlib import Path
|
||||
|
||||
import numpy as np
|
||||
from PIL import Image
|
||||
|
||||
|
||||
def main() -> int:
|
||||
if len(sys.argv) != 3:
|
||||
print(__doc__)
|
||||
return 1
|
||||
with Image.open(sys.argv[1]) as handle:
|
||||
image = handle.convert("L") if handle.mode in ("L", "1", "I;16") else handle.convert("RGB")
|
||||
array = np.asarray(image)
|
||||
if array.ndim == 2:
|
||||
array = array[..., None]
|
||||
Path(sys.argv[2]).write_bytes(np.ascontiguousarray(array).tobytes())
|
||||
print(array.shape[1], array.shape[0], array.shape[2])
|
||||
return 0
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
raise SystemExit(main())
|
||||
Reference in New Issue
Block a user