У меня есть CSV-файл с четырьмя столбцами:
- Путь к файлу изображения
- Функция изображения 1
- Функция изображения 2
- Метка для изображения
Я преобразовал файлы .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"
}}}
}