7
test.py
7
test.py
@@ -1,7 +0,0 @@
|
||||
from util import image_util
|
||||
|
||||
if __name__ == '__main__':
|
||||
image = r'test_img/PH20240813000294_3_081442_1.jpg'
|
||||
clas_result = image_util.parse_clarity(image)
|
||||
clas_result = next(clas_result)[0]
|
||||
print(clas_result)
|
||||
Binary file not shown.
|
Before Width: | Height: | Size: 1.1 MiB |
@@ -251,20 +251,3 @@ def combined(img1, img2):
|
||||
combined_img[:height1, :width1] = img1
|
||||
combined_img[:height2, width1:width1 + width2] = img2
|
||||
return combined_img
|
||||
|
||||
|
||||
def parse_clarity(image):
|
||||
"""
|
||||
判断图片清晰度
|
||||
:param image: 图片NumPy数组或文件路径
|
||||
:return: 判断结果及置信度
|
||||
"""
|
||||
clarity_result = [1, 0]
|
||||
model = PaddleClas(model_name="clarity_assessment")
|
||||
clas_result = model.predict(input_data=image)
|
||||
try:
|
||||
clas_result = next(clas_result)[0]
|
||||
clarity_result = [clas_result["class_ids"][0], clas_result["scores"][0]]
|
||||
except Exception as e:
|
||||
logging.error("获取图片清晰度失败", exc_info=e)
|
||||
return clarity_result
|
||||
|
||||
Reference in New Issue
Block a user