TFRecord文件的Pythonic工具
pytfrec的Python项目详细描述
pytfrec(pythonictfrecord工具)
此包包含用于轻松访问tfrec文件的函数。这个包的目的是提供一个简单的Python方法来处理tfrec文件,而不必处理Tensorflow文档中存在的技术复杂性。 目前这个包只包含处理图像数据集的函数
使用方法:
非常简单:
frompytfrec.imageimportget_datasetstfrecs_dir='path-to-tfrecs-dir'image_size=(64,64)batch_size=32validation_data_ratio=0.3datasets=get_datasets(tfrecs_dir,image_size,batch_size,validation_data_ratio,channels=3,labeled=True,ordered=False)# datasets is a dictionary which contain train and validation datasetstrain_dataset=datasets['train']validation_dataset=datasets['validation']# This datasets can be used directly in keras model.fit method
参考文献:
- 项目
标签: