Failed with my own tfrecords. about face-similarity HOT 2 CLOSED

The trace log is:
InvalidArgumentError Traceback (most recent call last)
in ()
4
5 print (type(test_dataset))
----> 6 for (batch,(Xi, Xj,label )) in enumerate(test_dataset):
7 print (Xi.shape)
8

~/anaconda3/lib/python3.6/site-packages/tensorflow/python/data/ops/iterator_ops.py in next(self)
522
523 def next(self): # For Python 3 compatibility
--> 524 return self.next()
525
526 def _next_internal(self):

~/anaconda3/lib/python3.6/site-packages/tensorflow/python/data/ops/iterator_ops.py in next(self)
553 """
554 try:
--> 555 return self._next_internal()
556 except errors.OutOfRangeError:
557 raise StopIteration

~/anaconda3/lib/python3.6/site-packages/tensorflow/python/data/ops/iterator_ops.py in _next_internal(self)
543 self._resource,
544 output_types=self._flat_output_types,
--> 545 output_shapes=self._flat_output_shapes)
546
547 return sparse.deserialize_sparse_tensors(

~/anaconda3/lib/python3.6/site-packages/tensorflow/python/ops/gen_dataset_ops.py in iterator_get_next_sync(iterator, output_types, output_shapes, name)
2181 else:
2182 message = e.message
-> 2183 _six.raise_from(_core._status_to_exception(e.code, message), None)
2184
2185

~/anaconda3/lib/python3.6/site-packages/six.py in raise_from(value, from_value)

InvalidArgumentError: Input to reshape is a tensor with 37686 values, but the requested shape has 37632
[[Node: image_reshape = Reshape[T=DT_UINT8, Tshape=DT_INT32](DecodeRaw, image_reshape/shape)]] [Op:IteratorGetNextSync]

from face-similarity.

I fixed it by comment out one line code:
#test_dataset = test_dataset.map(random_resize_and_crop)

It runs ok now. Thank you.

from face-similarity.