生成TFRecords的问题

Question

我正在尝试生成TFRecords。我正在使用Google Colab环境，但我认为这应该不是问题，因为我也尝试在本地运行它。我没有错误但也没有结果。

我在Google Colab中使用以下代码：

from google.colab import drive
drive.mount('/content/drive/')
!pip install tensorflow-object-detection-api
%cd /content/drive/My Drive/
!python generate_tfrecords.py --csv_input=Resistors/train_labels.csv --image_dir=Resistors/train --output_path=train.record

generate_tfrecords.py来自https://github.com/datitran/raccoon_dataset/blob/master/generate_tfrecord.py

但是我改变了分类。我真的很受困扰，而且我是TensorFlow的新手，感谢您的帮助！