diff --git a/test.py b/test.py deleted file mode 100644 index fc5cdc2..0000000 --- a/test.py +++ /dev/null @@ -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) diff --git a/test_img/PH20240813000294_3_081442_1.jpg b/test_img/PH20240813000294_3_081442_1.jpg deleted file mode 100644 index 2e20a7e..0000000 Binary files a/test_img/PH20240813000294_3_081442_1.jpg and /dev/null differ diff --git a/util/image_util.py b/util/image_util.py index d79c9f4..3a0082b 100644 --- a/util/image_util.py +++ b/util/image_util.py @@ -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