dzhwinter
7 years ago
parent
cc02353d10
commit
60f70b174d
@ -1,99 +0,0 @@
|
||||
|
||||
#include<windows.h>
|
||||
#include <fstream>
|
||||
#include "inference_icnet.h"
|
||||
#include <thread>
|
||||
#include <vector>
|
||||
#include <string>
|
||||
#include <iostream>
|
||||
|
||||
#include <sstream>
|
||||
using namespace std;
|
||||
|
||||
|
||||
template <class Type>
|
||||
Type stringToNum(const string& str)
|
||||
{
|
||||
istringstream iss(str);
|
||||
Type num;
|
||||
iss >> num;
|
||||
return num;
|
||||
}
|
||||
|
||||
void test_imgs() {
|
||||
void *h = init_predictor("./lb/__model__", "./lb/__params__", 0.3f, true, 0);
|
||||
|
||||
std::ifstream infile("new_file.list");
|
||||
std::ofstream ofs("./1.png.output.txt");
|
||||
|
||||
std::string temp_s;
|
||||
std::vector<std::string> all_files;
|
||||
while (!infile.eof()) {
|
||||
infile >> temp_s;
|
||||
all_files.push_back(temp_s);
|
||||
}
|
||||
// size_t file_num = all_files.size();
|
||||
infile.close();
|
||||
// =============read file list =============
|
||||
for (size_t f_k = 0; f_k < 1; f_k++) {
|
||||
// std::string path = "D:\\Paddle\\paddle\\fluid\\inference\\api\\demo_ci\\build\\Release\\";
|
||||
// std::ifstream in_img(path + all_files[f_k]);
|
||||
std::string mypath = "D:\\Paddle\\paddle\\fluid\\inference\\api\\demo_ci\\build\\Release\\1.png.txt";
|
||||
std::cout << "file" << mypath << std::endl;
|
||||
std::ifstream in_img(mypath);
|
||||
//std::cout << path + all_files[f_k] << std::endl;
|
||||
double temp_v;
|
||||
const int size = 3 * 449 * 581 * 1;
|
||||
float * data = new float[size];
|
||||
std::string value;
|
||||
|
||||
if (!in_img.is_open()) {
|
||||
cout << "open failed" << endl;
|
||||
}
|
||||
double sum_input = .0;
|
||||
for (auto i = 0; i < size; i++) {
|
||||
getline(in_img, value, '\n');
|
||||
double v = stringToNum<double>(value);
|
||||
data[i] = static_cast<float>(v);
|
||||
sum_input += v;
|
||||
}
|
||||
std::cout << "sum_input" << sum_input << std::endl;
|
||||
|
||||
in_img.close();
|
||||
const int SIZE = 449 * 581 * 1;
|
||||
int64_t * p = new int64_t[SIZE]();
|
||||
int out_size = 0;
|
||||
//memset(p, 0, size);
|
||||
predict(h, data, 3, 449, 581, &p, &out_size, 1);
|
||||
std::cout << "out_size = " << out_size << std::endl;
|
||||
|
||||
double out_sum = .0;
|
||||
for (auto i = 0; i < out_size / sizeof(int64_t); i++) {
|
||||
out_sum += p[i];
|
||||
ofs << p[i] << " ";
|
||||
}
|
||||
ofs.close();
|
||||
|
||||
std::cout << "inferece out sum" << out_sum << std::endl;
|
||||
delete p;
|
||||
}
|
||||
|
||||
destory_predictor(h);
|
||||
}
|
||||
|
||||
int main(int argc, char** argv) {
|
||||
//if (true) {
|
||||
// std::thread t1(func, init_predictor("./infer_model/__model__", "./infer_model/__params__", 0.1f, true, 0));
|
||||
// std::thread t2(func, init_predictor("./infer_model/__model__", "./infer_model/__params__", 0.1f, true, 0));
|
||||
// //std::thread t3(func, init_predictor("./infer_model/__model__", "./infer_model/__params__", 0.1f, true, 0));
|
||||
// //std::thread t4(func, init_predictor("./infer_model/__model__", "./infer_model/__params__", 0.1f, true, 0));
|
||||
// t1.join();
|
||||
// t2.join();
|
||||
// //t3.join();
|
||||
// //t4.join();
|
||||
// //Sleep(1);
|
||||
//}
|
||||
test_imgs();
|
||||
|
||||
return 0;
|
||||
}
|
||||
Loading…
Reference in new issue