python - 使用 tensorflow 训练对象检测分类器
问题描述
帮助我摆脱我无法克服的错误
使用以下命令后
python generate_tfrecord.py --csv_input=images/train_labels.csv --image_dir=images/train --output_path=train.record
我收到以下错误
Traceback (most recent call last):
File "C:\Users\asus\Anaconda3\lib\site-packages\tensorflow_core\python\pywrap_tensorflow.py", line 58, in <module>
from tensorflow.python.pywrap_tensorflow_internal import *
File "C:\Users\asus\Anaconda3\lib\site-packages\tensorflow_core\python\pywrap_tensorflow_internal.py", line 28, in <module>
_pywrap_tensorflow_internal = swig_import_helper()
File "C:\Users\asus\Anaconda3\lib\site-packages\tensorflow_core\python\pywrap_tensorflow_internal.py", line 24, in swig_import_helper
_mod = imp.load_module('_pywrap_tensorflow_internal', fp, pathname, description)
File "C:\Users\asus\Anaconda3\lib\imp.py", line 242, in load_module
return load_dynamic(name, filename, file)
File "C:\Users\asus\Anaconda3\lib\imp.py", line 342, in load_dynamic
return _load(spec)
ImportError: DLL load failed: The specified module could not be found.
During handling of the above exception, another exception occurred:
Traceback (most recent call last):
File "generate_tfrecord.py", line 18, in <module>
import tensorflow as tf
File "C:\Users\asus\Anaconda3\lib\site-packages\tensorflow\__init__.py", line 101, in <module>
from tensorflow_core import *
File "C:\Users\asus\Anaconda3\lib\site-packages\tensorflow_core\__init__.py", line 40, in <module>
from tensorflow.python.tools import module_util as _module_util
File "C:\Users\asus\Anaconda3\lib\site-packages\tensorflow\__init__.py", line 50, in __getattr__
module = self._load()
File "C:\Users\asus\Anaconda3\lib\site-packages\tensorflow\__init__.py", line 44, in _load
module = _importlib.import_module(self.__name__)
File "C:\Users\asus\Anaconda3\lib\importlib\__init__.py", line 127, in import_module
return _bootstrap._gcd_import(name[level:], package, level)
File "C:\Users\asus\Anaconda3\lib\site-packages\tensorflow_core\python\__init__.py", line 49, in <module>
from tensorflow.python import pywrap_tensorflow
File "C:\Users\asus\Anaconda3\lib\site-packages\tensorflow_core\python\pywrap_tensorflow.py", line 74, in <module>
raise ImportError(msg)
ImportError: Traceback (most recent call last):
File "C:\Users\asus\Anaconda3\lib\site-packages\tensorflow_core\python\pywrap_tensorflow.py", line 58, in <module>
from tensorflow.python.pywrap_tensorflow_internal import *
File "C:\Users\asus\Anaconda3\lib\site-packages\tensorflow_core\python\pywrap_tensorflow_internal.py", line 28, in <module>
_pywrap_tensorflow_internal = swig_import_helper()
File "C:\Users\asus\Anaconda3\lib\site-packages\tensorflow_core\python\pywrap_tensorflow_internal.py", line 24, in swig_import_helper
_mod = imp.load_module('_pywrap_tensorflow_internal', fp, pathname, description)
File "C:\Users\asus\Anaconda3\lib\imp.py", line 242, in load_module
return load_dynamic(name, filename, file)
File "C:\Users\asus\Anaconda3\lib\imp.py", line 342, in load_dynamic
return _load(spec)
ImportError: DLL load failed: The specified module could not be found.
Failed to load the native TensorFlow runtime.
See https://www.tensorflow.org/install/errors
for some common reasons and solutions.
Include the entire stack trace
above this error message when asking for help.
解决方案
我通过从这里下载和安装 Visual Studio 2017-2019 x86 和 x64 解决了这个问题:
其他解决方案是将 tensorflow 降级到 2.0:
pip install tensorflow==2.0
我有用于 CUDA 10.1 的 Python 3.7.6、CUDA 10.1 和 cuDNN,并且我遵循了以下要求:
如果这对您不起作用,请创建虚拟环境。
conda create -n tf-gpu tensorflow-gpu
conda activate tf-gpu
推荐阅读
- python - cv2.findContours 在黑色背景上找不到白色矩形
- node.js - 将 React 与 Electron 一起使用,以便仍然可以使用 Node.js 原生模块
- python - 指定方法的参数类型和返回类型有什么好处?
- amazon-web-services - 在 API Gateway 和 Lambda 函数中重复输入验证?
- html - css - 剪辑图像并添加边框以适合 div
- visual-studio-code - Xdebug 3.0 WSL2 和 VSCode - 有效的配置
- redis - 即使值存在,redis get 也返回 null
- javascript - 为什么日期比较 javascript 功能不起作用?
- javascript - 如何使用带有 Vue 的 HTML 5 验证添加错误消息
- docker - Docker以非交互方式设置用户密码