Tensorflow 生成tfrecords

Tfrecords 是Tensorflow的主流数据格式。可以用我写的脚本将图片方便的转化为tfrecords格式。
https://github.com/bxclib/generate_tfrecord
使用方式:
1 将图片放到root 文件夹下。(root 位置可以自己在settings.py里设定)
2 设置settings.py里的参数(通常要设置下root目录,matching_name字典和预处理图像的代码)
3 运行generate_tfrecords.py 或 generate_tfrecords_with_shuffle.py
三部曲,方便的生成tfrecords

猜你喜欢

转载自blog.csdn.net/xingce_cs/article/details/74779197