Структура TensorFlowRecord - PullRequest
0 голосов
/ 30 мая 2018

У меня есть CSV-файл с четырьмя столбцами:

  1. Путь к файлу изображения
  2. Функция изображения 1
  3. Функция изображения 2
  4. Метка для изображения

Я преобразовал файлы .csv в tensorflowRecords (каждые 100 строк, составляющих один файл TFrecord), и теперь я пытаюсь обучить модель с использованием файлов tfRecord.

Какой должна быть структура файла tfrecord? Должна ли она содержать только изображения и элементы, или мне нужно разделить метки и передать их в модель самостоятельно?Если это так, как мне сопоставить метки с соответствующими изображениями в файлах tensorflowRecord.В настоящее время структура tenorflow-примера выглядит следующим образом:

features {
    feature {
        key: "image"
        value {
            bytes_list {
                value: "\377\330\377\340\......." (too long to attach here)
    }}}
    feature {
        key: "feature1"
        value {
            float_list {
                value: 601456.0
    }}}
    feature {
        key: "feature2"
        value {
            float_list {
                value: 33321.0
    }}}
    feature {
        key: "label"
        value {
            bytes_list {
                value: "Tumor"
    }}}
}
...