File size: 604 Bytes
99fe130 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 | """用发布包识别一张白底黑字图片。"""
import argparse
from pathlib import Path
from inference import HCCRRecognizer
def main() -> None:
parser = argparse.ArgumentParser()
parser.add_argument("image", type=Path)
parser.add_argument("--model-dir", type=Path, default=Path("."))
parser.add_argument("-k", type=int, default=10)
args = parser.parse_args()
model = HCCRRecognizer.from_pretrained(args.model_dir)
for char, probability in model.predict_file(args.image, k=args.k):
print(f"{char}\t{probability:.2%}")
if __name__ == "__main__":
main()
|