KeyError为对象检测生成tf记录？

Question

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

当我在终端中输入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'

我该如何解决这个问题？

谢谢。

Answer 1

请尝试以下命令：

python2 generate_tfrecord.py --csv_input=images_train.csv --image_dir=images/train --output_path=train.record

Answer 2

请检查 csv 文件的列名。