Update Tensorflex.c

anshuman23 · Jul 3, 2018 · abee32d · abee32d
1 parent 4f49d07
commit abee32d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/c_src/Tensorflex.c b/c_src/Tensorflex.c
@@ -663,7 +663,7 @@ static ERL_NIF_TERM load_image_as_tensor(ErlNifEnv *env, int argc, const ERL_NIF
   jpeg_destroy_decompress(&cinfo);
   free(input_img);
 
-  const int size_alloc = output_size * sizeof(float);
+  const int size_alloc = output_size * sizeof(unsigned char);
   int64_t dims[3] = {width, height, num_pixels};
 
   tensor = TF_NewTensor(TF_UINT8, dims, 3, output, size_alloc, tensor_deallocator, 0);