В настоящее время я следую этому учебнику в Windows 10
и у меня есть требования. Но когда я прихожу на шаг
Преобразование из * .csv в * .record Я получаю ошибку
Это ошибка, которую я получаю
Я попробовал решения на github, изменив
classes_text.append(row['class'].encode('utf8')) до
classes_text.append(repr(row['class']))
в строке 80, однако это не сработало
Пример файла CSV
filename,width,height,class,xmin,ymin,xmax,ymax
6eadc7dfeb4fa757492ae20ff67edd50.jpg,dog,416,416,167,196,283,413
9d57e44f2028317e57cc4a3c2c836dd4.jpg,dog,416,416,56,37,101,111
python generate_tfrecord.py --label=dog--csv_input=C:\tensorflow_gpu\TensorFlow\models\space\training_demo\annotations\train_labels.csv --img_path=C:\tensorflow_gpu\TensorFlow\models\space\training_demo\images\train --output_path=C:\tensorflow_gpu\TensorFlow\models\space\training_demo\annotations\train.record
_sys.exit(main(argv))
File "generate_tfrecord_org.py", line 106, in main
tf_example = create_tf_example(group, path)
File "generate_tfrecord_org.py", line 80, in create_tf_example
classes_text.append(row['class'].encode('utf8'))
AttributeError: 'int' object has no attribute 'encode'
Я хочу, чтобы код вернул мне * .record файл, но он выдает ошибку и создает пустой * .record файл