@@ -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