sthalles / face-similarity Goto Github PK

Thank you for your share.
I got a problem when I run it with my own data.
I make bmp files to tfrecords with code below:

import tensorflow as tf
import cv2
import numpy as np

datalistfile = './images/data'
''' format of datalistfile
0 ./images/cgh_0.bmp ./images/cgh_10.bmp
0 ./images/cgh_10.bmp ./images/cgh_11.bmp
'''
dataf = open(datalistfile,'r')
writer = tf.python_io.TFRecordWriter("test.tfrecords")

def _int64_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]))

data = dataf.readlines()
for i in range(len(data)):
line = data[i]
line = line.strip()
line = line.split(" ")
Xi = tf.gfile.FastGFile(line[1], 'rb').read() # image data type is string
Xj = tf.gfile.FastGFile(line[2], 'rb').read() # image data type is string
image_shape = cv2.imread(line[1]).shape
width = image_shape[1]
height = image_shape[0]
label = int(line[0])
features = {
'Xi': _bytes_feature(Xi),
'Xj': _bytes_feature(Xj),
"label": _int64_feature(label),
"height": _int64_feature(height),
"width": _int64_feature(width)
}
example = tf.train.Example(features = tf.train.Features(feature=features))
writer.write(example.SerializeToString())
writer.close()

So , can you tell me how do you make your test_v2.tfrecords from images?

Hello i'm having some problems with your pynb :

'BatchDataset' object is not iterable

What is the tensorflow version needed ?