mirrors
/
PaddleClas
mirror of https://github.com/PaddlePaddle/PaddleClas.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

add gflags for cpp_shitu

pull/1437/head
dongshuilong 2021-11-09 07:13:18 +00:00
parent 5ef89df90e
commit c1a598f9ba
1 changed files with 16 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
deploy/cpp_shitu/src/main.cpp
View File

@ -27,6 +27,7 @@
#include <numeric>
#include <auto_log/autolog.h>
#include <gflags/gflags.h>
#include <include/cls.h>
#include <include/object_detector.h>
#include <include/vector_search.h>
@ -35,6 +36,9 @@
using namespace std;
using namespace cv;
DEFINE_string(config, "", "Path of yaml file");
DEFINE_string(c, "", "Path of yaml file");
void DetPredictImage(const std::vector<cv::Mat> &batch_imgs,
const std::vector<std::string> &all_img_paths,
const int batch_size, PaddleDetection::ObjectDetector *det,
@ -129,12 +133,21 @@ void DetPredictImage(const std::vector<cv::Mat> &batch_imgs,
}
int main(int argc, char **argv) {
if (argc != 2) {
std::cerr << "[ERROR] usage: " << argv[0] << " yaml_path\n";
google::ParseCommandLineFlags(&argc, &argv, true);
std::string yaml_path = "";
if (FLAGS_config == "" && FLAGS_c == "") {
std::cerr << "[ERROR] usage: " << std::endl
<< argv[0] << " -c $yaml_path" << std::endl
<< "or:" << std::endl
<< argv[0] << " -config $yaml_path" << std::endl;
exit(1);
} else if (FLAGS_config != "") {
yaml_path = FLAGS_config;
} else {
yaml_path = FLAGS_c;
}
YamlConfig config(argv[1]);
YamlConfig config(yaml_path);
config.PrintConfigInfo();
// initialize detector, rec_Model, vector_search