MXNet的数据格式转化为tensorflow训练用的数据格式

1 介绍

本文将mxNet的数据格式转化为tensorflow训练用的数据格式。

2 导入包

 

import mxnet as mx

import argparse

import PIL.Image

import io

import numpy as np

import cv2

import tensorflow as tf

import os

3 主函数

 

if __name__ == '__main__':     # define parameters     id2range = {}     data_shape = (3, 112, 112)     args = parse_args()    #idx_path放的是索引;args.bin_path放的是值，读取的话imgrec是一个IO的对象，可以理解为C++里面的类对象，类对象里面有很多成员；     # 比如有成员代表索引，成员代表索引对应的值得等。keys的len代表索引的数量     imgrec = mx.recordio.MXIndexedRecordIO(args.idx_path, args.bin_path, 'r')#索引长度5898396，也就是idx的长度     s = imgrec.read_idx(0)#读取最开始那个索引对应的key_value,不过s是二进制形式的     header, _ = mx.recordio.