code format for cls class_infer

pull/1557/head
dongshuilong 2021-12-14 14:48:22 +08:00
parent 3e30214702
commit e1716ded56
8 changed files with 364 additions and 360 deletions

Binary file not shown.

Before

Width:  |  Height:  |  Size: 85 KiB

After

Width:  |  Height:  |  Size: 73 KiB

View File

@ -125,7 +125,9 @@ public:
std::string label_save_dir;
void PrintConfigInfo();
void ReadLabelMap();
void ReadYamlConfig(const std::string &path);
};
} // namespace PaddleClas

View File

@ -209,7 +209,7 @@ cp ../configs/inference_cls.yaml tools/
根据[python预测推理](../../docs/zh_CN/inference_deployment/python_deploy.md)的`图像分类推理`部分修改好`tools`目录下`inference_cls.yaml`文件。`yaml`文件的参数说明详见[python预测推理](../../docs/zh_CN/inference_deployment/python_deploy.md)。
请根据实际存放文件,修改好`Global.infer_imgs`、`Global.inference_model_dir`等参数。
请根据实际存放文件,修改好`Global.infer_imgs`、`Global.inference_model_dir`、`PostProcess.Topk.topk`、`PostProcess.Topk.class_id_map_file`等参数。
#### 2.3.2 执行

View File

@ -36,8 +36,10 @@ using namespace std;
using namespace cv;
using namespace PaddleClas;
DEFINE_string(config, "", "Path of yaml file");
DEFINE_string(c, "", "Path of yaml file");
DEFINE_string(config,
"", "Path of yaml file");
DEFINE_string(c,
"", "Path of yaml file");
int main(int argc, char **argv) {
google::ParseCommandLineFlags(&argc, &argv, true);