f171796995d17e3db4b4752a116e532d499dc91f,src/caffe/pyutil/convert.py,,array_to_datum,#,20

Before Change

def array_to_datum(arr):
  if arr.ndim != 3:
    raise ValueError("Incorrect array shape.")
  if arr.dtype != np.uint8:
    raise TypeError("Input array has to be of type uint8.")
  datum = caffe_pb2.Datum()
  datum.channels, datum.height, datum.width = arr.shape
  datum.data = arr.tostring()
  return datum

After Change

  if arr.dtype == np.uint8:
    datum.data = arr.tostring()
  else:
    datum.float_data.extend(arr.flat)
  return datum

Instances