Мой csv-файл содержал imagenames с расширением jpg, и у меня все еще была эта ошибка.Я попытался решить ее с помощью:
python3 generate_tf_record.py --csv_input=data/train_labels.csv --output_path=train.record
python3 generate_tf_record.py --csv_input=data/test_labels.csv --output_path=test.record
Все изображения были в одной папке и помечены флажками, как показано ниже:
flags.DEFINE_string('csv_input', '','data/train_labels.csv')
flags.DEFINE_string('output_path','', 'train.record')
flags.DEFINE_string('image_dir', '', 'images')
Проблема была решена, когда я скопировал сохраненный файл записи из основногопапка в папку данных.