KeyError生成用于对象检测的tf记录？

Question

我正在按照本教程/随附的代码[https://gilberttanner.com/blog/creating-your-own-objectdetector]在tensorflow中运行对象检测API。>

当我在终端中键入python2 generate_tfrecord.py --csv_input=images_train.csv --image_dir=images\train --output_path=train.record时，我看到在该目录中创建了一个文件train.record，但是我也收到以下错误消息：

Traceback (most recent call last):
  File "generate_tfrecord.py", line 107, in <module>
    tf.app.run()
  File "/usr/local/lib/python2.7/site-packages/tensorflow_core/python/platform/app.py", line 40, in run
    _run(main=main, argv=argv, flags_parser=_parse_flags_tolerate_undef)
  File "/usr/local/lib/python2.7/site-packages/absl/app.py", line 299, in run
    _run_main(main, args)
  File "/usr/local/lib/python2.7/site-packages/absl/app.py", line 250, in _run_main
    sys.exit(main(argv))
  File "generate_tfrecord.py", line 96, in main
    grouped = split(examples, 'filename')
  File "generate_tfrecord.py", line 47, in split
    gb = df.groupby(group)
  File "/Users/sofiatomov/Library/Python/2.7/lib/python/site-packages/pandas/core/generic.py", line 6665, in groupby
    observed=observed, **kwargs)
  File "/Users/sofiatomov/Library/Python/2.7/lib/python/site-packages/pandas/core/groupby/groupby.py", line 2152, in groupby
    return klass(obj, by, **kwds)
  File "/Users/sofiatomov/Library/Python/2.7/lib/python/site-packages/pandas/core/groupby/groupby.py", line 599, in __init__
    mutated=self.mutated)
  File "/Users/sofiatomov/Library/Python/2.7/lib/python/site-packages/pandas/core/groupby/groupby.py", line 3291, in _get_grouper
    raise KeyError(gpr)
KeyError: 'filename'
我该如何解决？

谢谢。

我正在尝试按照本教程/随附的代码在tensorflow中运行对象检测API：https://gilberttanner.com/blog/creating-your-own-objectdetector当我键入python2 ...

Answer 1

请尝试以下命令：