测试图片清晰度模型效果

tool/check_clarity_model.py

@@ -0,0 +1,43 @@
+import pandas as pd
+
+from db import MysqlSession
+from db.mysql import ZxPhrec
+
+
+def check_unclarity():
+    file_path = r'D:\Echo\Downloads\unclare.xlsx'  # 将 your_file.xlsx 替换为你的文件路径
+    df = pd.read_excel(file_path)
+
+    # 循环读取 pk_phrec 列
+    output_file_path = 'check_unclarity_result.txt'
+    with open(output_file_path, 'w') as f:
+        for pk_phrec in df['pk_phrec']:
+            session = MysqlSession()
+            phrec = session.query(ZxPhrec.pk_phrec, ZxPhrec.cfjaddress2, ZxPhrec.unsharp_flag).filter(
+                ZxPhrec.pk_phrec == pk_phrec
+            ).one_or_none()
+            session.close()
+            if phrec and phrec.unsharp_flag == 0:
+                f.write(f"{phrec.pk_phrec} {phrec.cfjaddress2}\n")
+
+
+def check_clarity():
+    file_path = r'D:\Echo\Downloads\unclare.xlsx'
+    df = pd.read_excel(file_path)
+    session = MysqlSession()
+    phrecs = (session.query(ZxPhrec.pk_phrec, ZxPhrec.cfjaddress2, ZxPhrec.unsharp_flag)
+              .filter(ZxPhrec.pk_phrec >= 30810206)
+              .filter(ZxPhrec.pk_phrec <= 31782247)
+              .filter(ZxPhrec.unsharp_flag == 1)
+              .all())
+    session.close()
+
+    output_file_path = 'check_clarity_result.txt'
+    with open(output_file_path, 'w') as f:
+        for phrec in phrecs:
+            if phrec.pk_phrec not in df['pk_phrec'].to_list():
+                f.write(f"{phrec.pk_phrec} {phrec.cfjaddress2}\n")
+
+
+if __name__ == '__main__':
+    check_clarity()