csv转tfrecord

#!/usr/bin/env python
# -*- coding:utf-8 -*-

# generate_tfrecord.py

# -*- coding: utf-8 -*-


"""
Usage:
  # From tensorflow/models/
  # Create train data:
  python generate_tfrecord.py --csv_input=data/tv_vehicle_labels.csv  --output_path=train.record
  # Create test data:
  python generate_tfrecord.py --csv_input=data/test_labels.csv  --output_path=test.record
"""


import io
import os
from collections import namedtuple

import pandas as pd
import tensorflow as tf
from PIL import Image
from object_detection.utils import dataset_util

os.chdir(r'E:\AI\打标')

flags = tf.app.flags
flags.DEFINE_string('csv_input', r'E:\AI\打标\train.csv', 'Path to the CSV input')
flags.DEFINE_string('output_path', r'E:\AI\打标\train.record', 'Path to output TFRecord')
FLAGS = flags.FLAGS


# TO-DO replace this with label map
def class_text_to_int(row_label):
    if row_label == 'person':     # 需改动
        return 1
    else:
        None


def split(df, group):
    data = namedtuple(