tenseorflow2.0目标检测 csv to tfcord

最新推荐文章于 2023-09-18 17:55:42 发布

hacker_wooder

最新推荐文章于 2023-09-18 17:55:42 发布

阅读量536

点赞数

文章标签： csv python

本文链接：https://blog.csdn.net/weixin_35873939/article/details/103688215

版权

这篇博客记录了使用TensorFlow 2.0将CSV格式的数据转换为TFRecord，适用于目标检测任务的过程。内容将涵盖从CSV文件读取数据、预处理、以及转换为TFRecord格式的详细步骤。

摘要由CSDN通过智能技术生成

先做记录之后修改详细内容

from __future__ import division
from __future__ import print_function
from __future__ import absolute_import

import os
import io
import pandas as pd
import tensorflow as tf
from PIL import Image
#from object_detection.utils import dataset_util
from collections import namedtuple, OrderedDict

def int64_feature(value):
  return tf.train.Feature(int64_list=tf.train.Int64List(value=[value]))


def int64_list_feature(value):
  return tf.train.Feature(int64_list=tf.train.Int64List(value=value))


def bytes_feature(value):
  return tf.train.Feature(bytes_list=tf.train.BytesList(value=[value]))


def bytes_list_feature(value):
  return tf.train.Feature(bytes_list=tf.train.BytesList(value=value))


def float_list_feature(value):
  return tf.train