File tree Expand file tree Collapse file tree 12 files changed +37
-83
lines changed
Expand file tree Collapse file tree 12 files changed +37
-83
lines changed Load Diff This file was deleted.
Original file line number Diff line number Diff line change 1- add_executable (coco_example main.cpp ../../src/Model.cpp ../../src/Tensor.cpp)
2- find_package (OpenCV REQUIRED )
3- target_include_directories (coco_example PRIVATE ../../include )
4- target_link_libraries (coco_example ${LIBS} ${OpenCV_LIBS} )
1+ cmake_minimum_required (VERSION 3.10)
2+ project (example)
3+
4+ set (CMAKE_CXX_STANDARD 17)
5+ add_executable (example main.cpp ../../src/Model.cpp ../../src/Tensor.cpp)
6+ find_package ( OpenCV REQUIRED )
7+ target_include_directories (example PRIVATE ../../include )
8+ target_link_libraries (example -ltensorflow ${OpenCV_LIBS} )
Original file line number Diff line number Diff line change 22// Created by sergio on 16/05/19.
33//
44
5- #include " ../../src /Model.h"
6- #include " ../../src /Tensor.h"
5+ #include " ../../include /Model.h"
6+ #include " ../../include /Tensor.h"
77#include < opencv2/opencv.hpp>
88#include < numeric>
99#include < iomanip>
Original file line number Diff line number Diff line change 1+ cmake_minimum_required (VERSION 3.10)
2+ project (example)
3+
14set (CMAKE_CXX_STANDARD 17)
2- add_executable (load_model_example main.cpp ../../src/Model.cpp ../../src/Tensor.cpp)
3- target_include_directories (load_model_example PRIVATE ../../include )
4- target_link_libraries (load_model_example ${LIBS} )
5+ add_executable (example main.cpp ../../src/Model.cpp ../../src/Tensor.cpp)
6+ target_include_directories (example PRIVATE ../../include )
7+ target_link_libraries (example -ltensorflow )
Original file line number Diff line number Diff line change 22// Created by sergio on 16/05/19.
33//
44
5- #include " ../../src /Model.h"
6- #include " ../../src /Tensor.h"
5+ #include " ../../include /Model.h"
6+ #include " ../../include /Tensor.h"
77
88#include < numeric>
99#include < iomanip>
Original file line number Diff line number Diff line change 1- find_package (OpenCV REQUIRED )
2- add_executable (nmist_example main.cpp ../../src/Model.cpp ../../src/Tensor.cpp)
3- target_include_directories (nmist_example PRIVATE ../../include )
4- target_link_libraries (nmist_example ${LIBS} ${OpenCV_LIBS} )
1+ cmake_minimum_required (VERSION 3.10)
2+ project (example)
3+
4+ set (CMAKE_CXX_STANDARD 17)
5+ find_package ( OpenCV REQUIRED )
6+ add_executable (example main.cpp ../../src/Model.cpp ../../src/Tensor.cpp)
7+ target_include_directories (example PRIVATE ../../include )
8+ target_link_libraries (example -ltensorflow ${OpenCV_LIBS} )
Original file line number Diff line number Diff line change 22// Created by sergio on 12/05/19.
33//
44
5- #include " ../../src /Model.h"
6- #include " ../../src /Tensor.h"
5+ #include " ../../include /Model.h"
6+ #include " ../../include /Tensor.h"
77#include < opencv2/opencv.hpp>
88#include < algorithm>
99#include < iterator>
1212int main () {
1313
1414 // Create model
15- Model m (" model.pb" );
16- m.restore (" checkpoint/train.ckpt" );
15+ Model m (" ../ model.pb" );
16+ m.restore (" ../ checkpoint/train.ckpt" );
1717
1818 // Create Tensors
1919 Tensor input (m, " input" );
@@ -24,7 +24,7 @@ int main() {
2424 cv::Mat img, scaled;
2525
2626 // Read image
27- img = cv::imread (" images/" +std::to_string (i)+" .png" );
27+ img = cv::imread (" ../ images/" +std::to_string (i)+" .png" );
2828
2929 // Scale image to range 0-1
3030 img.convertTo (scaled, CV_64F, 1 .f /255 );
Original file line number Diff line number Diff line change 55#ifndef CPPFLOW_MODEL_H
66#define CPPFLOW_MODEL_H
77
8+ #include < cstring>
9+ #include < algorithm>
810#include < string>
911#include < vector>
1012#include < iostream>
1113#include < fstream>
1214#include < tuple>
13- #include " c_api.h"
15+ #include < tensorflow/c/ c_api.h>
1416#include " Tensor.h"
1517
1618class Tensor ;
Original file line number Diff line number Diff line change 1111#include < algorithm>
1212#include < numeric>
1313#include < cstring>
14- #include " c_api.h"
14+ #include < tensorflow/c/ c_api.h>
1515#include " Model.h"
1616
1717class Model ;
You can’t perform that action at this time.
0 commit comments