解决 'cnocr' 库 ImportError: cannot import name 'ConvBNActivation' 错误
在使用 'cnocr' 库进行 OCR 识别时,遇到了如下错误:
#图片名称
file_name = './data/cn_name.png'
ocr = CnOcr()
result = ocr.ocr(file_name)
print(f'Predicted Chars:{result}')
报错信息:
D:\pythonProject\venv\Scripts\python.exe D:\pythonProject\OCR-PRJ\ocr-remote.py
Traceback (most recent call last):
File "D:\pythonProject\OCR-PRJ\ocr-remote.py", line 1, in <module>
from cnocr import CnOcr
File "D:\pythonProject\venv\Lib\site-packages\cnocr\__init__.py", line 21, in <module>
from .cn_ocr import CnOcr, gen_model
File "D:\pythonProject\venv\Lib\site-packages\cnocr\cn_ocr.py", line 31, in <module>
from cnocr.models.ocr_model import OcrModel
File "D:\pythonProject\venv\Lib\site-packages\cnocr\models\ocr_model.py", line 34, in <module>
from .mobilenet import gen_mobilenet_v3
File "D:\pythonProject\venv\Lib\site-packages\cnocr\models\mobilenet.py", line 25, in <module>
from torchvision.models.mobilenetv2 import ConvBNActivation
ImportError: cannot import name 'ConvBNActivation' from 'torchvision.models.mobilenetv2' (D:\pythonProject\venv\Lib\site-packages\torchvision\models\mobilenetv2.py)
Process finished with exit code 1
该错误表明无法从 torchvision.models.mobilenetv2 中导入 ConvBNActivation。这通常是由于 PyTorch 或 torchvision 版本不兼容导致的。
解决方案:
-
更新 PyTorch 和 torchvision: 尝试更新 PyTorch 和 torchvision 到最新版本,并确保它们与您的系统和 cnocr 库兼容。
-
降低 cnocr 库版本: 如果更新 PyTorch 和 torchvision 无法解决问题,可以尝试降低 cnocr 库版本。您可以使用 pip uninstall cnocr 和 pip install cnocr==[版本号] 来卸载和安装特定版本的 cnocr 库。
注意:
- 确保您的 Python 环境中安装了 PyTorch 和 torchvision。
- 在更新或降低库版本后,请重新启动您的 Python 环境以使更改生效。
如果您仍然遇到问题,请在评论区中描述您的环境配置和错误信息,以便我提供更具体的解决方案。
原文地址: https://www.cveoy.top/t/topic/mv1b 著作权归作者所有。请勿转载和采集!