fix:package

rocker9527 · Sep 5, 2024 · 60ff66a · 60ff66a
1 parent c864c8e
commit 60ff66a
Show file tree

Hide file tree

Showing 4 changed files with 14 additions and 12 deletions.
diff --git a/hivision/__init__.py b/hivision/__init__.py
@@ -1 +1,4 @@
+from .creator import IDCreator, Params as IDParams, Result as IDResult
 
+
+__all__ = ["IDCreator", "IDParams", "IDResult", "utils", "error"]
diff --git a/utils/image_utils.py → hivision/utils.py b/utils/image_utils.py → hivision/utils.py
@@ -1,3 +1,12 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+r"""
+@DATE: 2024/9/5 21:52
+@File: utils.py
+@IDE: pycharm
+@Description:
+    hivision提供的工具函数
+"""
 from PIL import Image
 import io
 import numpy as np

diff --git a/test/create_id_photo.py b/test/create_id_photo.py
@@ -18,5 +18,5 @@
 image = cv2.imread(image_path)
 creator = IDCreator()
 result = creator(image)
-cv2.imwrite(os.path.join(output_dir, "result.jpg"), result.standard)
-cv2.imwrite(os.path.join(output_dir, "result_hd.jpg"), result.hd)
+cv2.imwrite(os.path.join(output_dir, "result.png"), result.standard)
+cv2.imwrite(os.path.join(output_dir, "result_hd.png"), result.hd)
diff --git a/utils/__init__.py b/utils/__init__.py
Original file line number	Diff line number	Diff line change
		@@ -1 +1,4 @@
		from .creator import IDCreator, Params as IDParams, Result as IDResult


		__all__ = ["IDCreator", "IDParams", "IDResult", "utils", "error"]