137 lines
4.2 KiB
Python
137 lines
4.2 KiB
Python
#!/usr/bin/env python3
|
||
# -*- coding: utf-8 -*-
|
||
"""
|
||
简化OCR测试脚本
|
||
"""
|
||
|
||
import cv2
|
||
import json
|
||
from pathlib import Path
|
||
|
||
def test_paddleocr():
|
||
"""测试PaddleOCR基本功能"""
|
||
print("🔧 测试PaddleOCR...")
|
||
|
||
try:
|
||
from paddleocr import PaddleOCR
|
||
|
||
# 初始化OCR
|
||
ocr = PaddleOCR(
|
||
use_textline_orientation=True,
|
||
lang='ch'
|
||
)
|
||
print("✅ PaddleOCR初始化成功")
|
||
|
||
# 测试视频文件
|
||
video_path = "/root/autodl-tmp/hot_video_analyse/video2audio/sample_demo_6.mp4"
|
||
|
||
# 读取一帧进行测试
|
||
cap = cv2.VideoCapture(video_path)
|
||
ret, frame = cap.read()
|
||
cap.release()
|
||
|
||
if not ret:
|
||
print("❌ 无法读取视频帧")
|
||
return False
|
||
|
||
print("📸 成功读取视频帧")
|
||
|
||
# 进行OCR识别 - 使用predict方法
|
||
try:
|
||
result = ocr.predict(frame)
|
||
except:
|
||
# 如果predict不工作，尝试ocr方法
|
||
result = ocr.ocr(frame)
|
||
|
||
print(f"🔍 OCR结果类型: {type(result)}")
|
||
|
||
# 处理结果
|
||
if result:
|
||
if isinstance(result, list) and len(result) > 0:
|
||
detections = result[0] if isinstance(result[0], list) else result
|
||
print(f"🔍 OCR结果: {len(detections)} 个检测")
|
||
|
||
for i, detection in enumerate(detections[:3]): # 只显示前3个
|
||
if detection and len(detection) >= 2:
|
||
try:
|
||
bbox, text_info = detection
|
||
if isinstance(text_info, tuple) and len(text_info) >= 2:
|
||
text, confidence = text_info
|
||
print(f" {i+1}. 文本: '{text}' 置信度: {confidence:.3f}")
|
||
else:
|
||
print(f" {i+1}. 检测到文本但格式异常: {text_info}")
|
||
except Exception as e:
|
||
print(f" {i+1}. 处理检测结果时出错: {e}")
|
||
else:
|
||
print("🔍 OCR结果为空或格式异常")
|
||
else:
|
||
print("🔍 OCR结果为None")
|
||
|
||
return True
|
||
|
||
except Exception as e:
|
||
print(f"❌ PaddleOCR测试失败: {e}")
|
||
import traceback
|
||
print(f"详细错误: {traceback.format_exc()}")
|
||
return False
|
||
|
||
def test_easyocr():
|
||
"""测试EasyOCR基本功能"""
|
||
print("\n🔧 测试EasyOCR...")
|
||
|
||
try:
|
||
import easyocr
|
||
|
||
# 初始化OCR
|
||
reader = easyocr.Reader(['ch_sim', 'en'])
|
||
print("✅ EasyOCR初始化成功")
|
||
|
||
# 测试视频文件
|
||
video_path = "/root/autodl-tmp/hot_video_analyse/video2audio/sample_demo_6.mp4"
|
||
|
||
# 读取一帧进行测试
|
||
cap = cv2.VideoCapture(video_path)
|
||
ret, frame = cap.read()
|
||
cap.release()
|
||
|
||
if not ret:
|
||
print("❌ 无法读取视频帧")
|
||
return
|
||
|
||
print("📸 成功读取视频帧")
|
||
|
||
# 进行OCR识别
|
||
result = reader.readtext(frame)
|
||
|
||
print(f"🔍 OCR结果: {len(result)} 个检测")
|
||
|
||
for i, detection in enumerate(result[:3]): # 只显示前3个
|
||
bbox, text, confidence = detection
|
||
print(f" {i+1}. 文本: '{text}' 置信度: {confidence:.3f}")
|
||
|
||
return True
|
||
|
||
except Exception as e:
|
||
print(f"❌ EasyOCR测试失败: {e}")
|
||
return False
|
||
|
||
def main():
|
||
"""主函数"""
|
||
print("🧪 简化OCR功能测试")
|
||
print("=" * 40)
|
||
|
||
paddle_ok = test_paddleocr()
|
||
easy_ok = test_easyocr()
|
||
|
||
print("\n" + "=" * 40)
|
||
print("📊 测试结果:")
|
||
print(f" PaddleOCR: {'✅ 成功' if paddle_ok else '❌ 失败'}")
|
||
print(f" EasyOCR: {'✅ 成功' if easy_ok else '❌ 失败'}")
|
||
|
||
if paddle_ok or easy_ok:
|
||
print("\n🎉 至少有一个OCR引擎工作正常，可以继续使用！")
|
||
else:
|
||
print("\n⚠️ 所有OCR引擎都有问题，请检查安装")
|
||
|
||
if __name__ == "__main__":
|
||
main() |