1 实现思路

基本介绍： 思路来源于atlas,atc(模型转换工具)，该工具对命令支持众多，且命令支持盲录入，支持基本的–help命令查看所有命令参数的解释，因此自己仿照实现，已解决工作当中的通用性工具使用不方便、难以产品化的问题。
atc 工具的应用命令如下：

atc --framework=5 --model=./yolov5l_onnx --input_format=NCHW --input_shape="images:1,3,960,960" --output_type=FP32 --output=./yolov5l_onnx --log=error --precision_mode=allow_fp32_to_fp16 --soc_version=Ascend310P3

下图为atc工具help命令效果：

2 主要功能

• 支持跨平台，代码简单，轻松实现移植到linux
• 支持命令的无序输入
• 支持 –help 命令查看 ./cmdParser --help
• 支持错误命令检查告警
• 支持直接编辑完成命令扩展

3 效果展示

3.1 直接运行代码图

3.2help命令执行效果图

3.3命令行执行命令解析效果图

4 代码实现

#include <iostream>
#include <map>
#include <vector>
#include <string>
#include <sstream>
#include <stdio.h>//命令解析方法
std::map<std::string, std::string> parseCommand(const std::string& command) {std::map<std::string, std::string> parameters;std::istringstream iss(command);std::string token;while (std::getline(iss, token, ' ')) {size_t pos = token.find('=');if (pos != std::string::npos) {std::string key = token.substr(0, pos);std::string value = token.substr(pos + 1);parameters[key] = value;}}return parameters;
}//命令库，直接可以复制编辑
std::map<std::string, std::string> cmdlib = {{"--framework","<number> Specify the framework version"},{"--model","<path> Specify the model file path"},{"--input_format","<format> Specify the input format"},{"--input_shape","<shape> Specify the input shape"},{"--output_type","<type> Specify the output type"},{"--output","<path> Specify the output file path"},{"--log","<level> Specify the log level"},{"--precision_mode","<mode> Specify the precision mode"},{"--soc_version","<version> Specify the SoC version"},{"--help","Display this help and exit"}
};//help展示所有命令
void displayHelp()
{std::cout << "Usage: command [options]\n"<< "Options:\n";for (const auto& p : cmdlib){//命令对齐排版printf("%-30s%-50s\n", p.first.c_str(), p.second.c_str());}
}int main(int argc, char* argv[]) {//解析help命令if (argc == 2 && std::string(argv[1]).find("--help") != std::string::npos) {displayHelp();return 0;}//将所有输入的命令都连成一个字符串std::string command;for (int i = 1; i < argc; ++i) { // Start from 1 to skip the program namecommand += argv[i];if (i < argc - 1) { // Add a space between arguments, but not after the last onecommand += " ";}}std::cout << "The combined command is: " << command << std::endl;//错误 命令测试 command = "--input_shape=\"images:1,3,960,960\" --output_type=FP32 --output=./yolov5l_onnx --log=error --precision_mode=allow_fp32_to_fp16 --soc_version=Ascend310P3  --framework=5 --model=./yolov5l --input_format=NCHW";//正确 测试命令 command = "--input_shape=\"images:1,3,960,960\" --output_type=FP32 --output=./yolov5l_onnx --log=error --precision_mode=allow_fp32_to_fp16 --soc_version=Ascend310P3  --framework=5 --model=./yolov5l --input_format=NCHW";//解析输入的字符串命令auto parameters = parseCommand(command);if (!parameters.empty()){for (const auto& p : parameters){//检查输入的命令是否存在非法命令auto itor = cmdlib.find(p.first);if (itor == cmdlib.end()){std::cerr << "Error: Please Check Invalid command:  " << p.first << std::endl;return 0;}}//打印解析输入命令得到的数据结构for (const auto& p : parameters){std::cout << p.first << " = " << p.second << std::endl;}}return 0;
}

5 代码下载

cmdParser