小编9hy*_*hyo的帖子

使用tensorflow-gpu进行对象检测时出现问题

我正在阅读youtube教程:https://www.youtube.com/watch？v = RMpfk6eYxJA

我正在尝试使用tensorflow-gpu和虚拟环境来检测对象.

我在系统环境变量中添加了python,cuda,tensorflow,并且还制作了带标签的训练模型.

我使用xml_to_csv.py将xml标签转换为csv.

问题是当我尝试使用generate_tfrecord.py生成tfrecord时,会出现该错误.请帮忙

这是代码

(tensorflow) C:\Users\ice305\tensorflow\models\research\object_detection>python generate_tfrecord.py --csv_input=images\train_labels.csv --image_dir=images\train --output_path=train.record
Traceback (most recent call last):
  File "generate_tfrecord.py", line 17, in <module>
    import tensorflow as tf
  File "C:\Anaconda3\envs\tensorflow\lib\site-packages\tensorflow\__init__.py", line 22, in <module>
    from tensorflow.python import pywrap_tensorflow  # pylint: disable=unused-import
  File "C:\Anaconda3\envs\tensorflow\lib\site-packages\tensorflow\python\__init__.py", line 81, in <module>
    from tensorflow.python import keras
  File "C:\Anaconda3\envs\tensorflow\lib\site-packages\tensorflow\python\keras\__init__.py", line 24, in <module>
    from tensorflow.python.keras import activations
  File "C:\Anaconda3\envs\tensorflow\lib\site-packages\tensorflow\python\keras\activations\__init__.py", line 22, in <module>
    from tensorflow.python.keras._impl.keras.activations import elu
  File "C:\Anaconda3\envs\tensorflow\lib\site-packages\tensorflow\python\keras\_impl\keras\__init__.py", line 21, …