saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-30 20:18:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix fns candy build error on Linux

Browse source
This commit is contained in:
Changming Sun 2019-10-28 17:38:44 -07:00
parent b0359b5622
commit d1096b524f
1 changed files with 3 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
samples/c_cxx/fns_candy_style_transfer/fns_candy_style_transfer.c
View file

@ -140,6 +140,7 @@ static int write_tensor_to_png_file(OrtValue* tensor, const char* output_file) {
static void usage() { printf("usage: <model_path> <input_file> <output_file> [cpu|cuda|dml] \n"); }
#ifdef _WIN32
static char* convert_string(const wchar_t* input) {
size_t src_len = wcslen(input) + 1;
if (src_len > INT_MAX) {
@ -154,6 +155,7 @@ static char* convert_string(const wchar_t* input) {
assert(len == r);
return ret;
}
#endif
int run_inference(OrtSession* session, const ORTCHAR_T* input_file, const ORTCHAR_T* output_file) {
size_t input_height;
@ -164,6 +166,7 @@ int run_inference(OrtSession* session, const ORTCHAR_T* input_file, const ORTCHA
char* output_file_p = convert_string(output_file);
char* input_file_p = convert_string(input_file);
#else
char* output_file_p = output_file;
char* input_file_p = input_file;
#endif
if (read_png_file(input_file_p, &input_height, &input_width, &model_input, &model_input_ele_count) != 0) {