图片文件涉密内容检测 - 使用文字识别技术自动生成检查报告
图片文件涉密内容检测 - 使用文字识别技术自动生成检查报告
本文介绍使用Python结合OpenCV和Tesseract-OCR库,对文件夹中的图片文件进行文字识别,并检测其中是否包含涉密内容。程序自动生成检查报告,包含图片数量统计、涉密图片目录地址及文件名等信息。
可行方案:
- 使用Python编程语言,结合第三方库如OpenCV、Tesseract-OCR等实现对图片的读取和文字识别功能。
- 遍历指定文件夹中的所有图片文件,对每个文件进行文字识别并检查其中是否存在涉密信息。
- 将检查结果存储在一个列表中,包括文件名、目录地址和是否含有涉密信息等信息。
- 统计各个类型的图片数量和含有涉密信息的图片数量,最终生成检查报告。
Python实现代码示例:
import os
import cv2
import pytesseract
# 设置Tesseract-OCR的安装路径
pytesseract.pytesseract.tesseract_cmd = r'C:\Program Files\Tesseract-OCR\tesseract.exe'
# 定义敏感词列表
sensitive_words = ['机密', '保密', '秘密']
# 定义检查结果列表
result = []
# 遍历指定文件夹中的所有图片文件
for root, dirs, files in os.walk('path/to/folder'):
for file in files:
# 判断文件是否为图片文件
if file.endswith('.jpg') or file.endswith('.png') or file.endswith('.bmp'):
# 读取图片文件
img = cv2.imread(os.path.join(root, file))
# 将图片转为灰度图像
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
# 对灰度图像进行二值化处理
_, binary = cv2.threshold(gray, 0, 255, cv2.THRESH_BINARY_INV | cv2.THRESH_OTSU)
# 对二值化图像进行文字识别
text = pytesseract.image_to_string(binary, lang='chi_sim')
# 判断识别结果中是否存在敏感词
if any(word in text for word in sensitive_words):
# 将检查结果添加到列表中
result.append({'directory': root, 'filename': file, 'is_sensitive': True})
else:
result.append({'directory': root, 'filename': file, 'is_sensitive': False})
# 统计各个类型的图片数量和含有涉密信息的图片数量
total_count = len(result)
jpg_count = sum(1 for r in result if r['filename'].endswith('.jpg'))
png_count = sum(1 for r in result if r['filename'].endswith('.png'))
bmp_count = sum(1 for r in result if r['filename'].endswith('.bmp'))
sensitive_count = sum(1 for r in result if r['is_sensitive'])
# 生成检查报告
report = f'Total: {total_count}
JPG: {jpg_count}
PNG: {png_count}
BMP: {bmp_count}
Sensitive: {sensitive_count}
Sensitive images:
'
for r in result:
if r['is_sensitive']:
report += f'{r['directory']}/{r['filename']}
'
print(report)
运行以上代码,即可对指定文件夹中的所有图片进行文字识别和涉密信息检查,并生成检查报告。
注意:
- 需确保已安装OpenCV和Tesseract-OCR库,并设置Tesseract-OCR的安装路径。
- 敏感词列表可根据实际需求进行调整。
- 代码示例中使用简化的文字识别方法,实际应用中可能需要更复杂的预处理和识别方法以提高识别准确率。
- 代码仅供参考,实际应用中需要根据具体情况进行修改和完善。
原文地址: http://www.cveoy.top/t/topic/oHLK 著作权归作者所有。请勿转载和采集!