tensorrtx加速推理后，如何把80个分类转换为自己的分类个数

1，主要修改的是det.cpp文件，我将分类作为配置文件拿出来，修改一下
在yolov5.det里添加函数：

std::vector read_gun_classes(std::string file_name) {
    std::vector classes;
    std::ifstream ifs(file_name, std::ios::in);
    if (!ifs.is_open()) {
        std::cerr << file_name << " is not found, pls refer to README and download it." << std::endl;
        assert(0);
    }
    std::string s;
    while (std::getline(ifs, s)) {
        classes.push_back(s);
    }
    ifs.close();
    return classes;
}

重载修改：draw_bbox函数
draw_bbox(img_batch, res_batch,gun_classes);
2，最终就可以在class.txt里修改分类的内容了。

代码不全，作为备注记录，全的参考代码上传到网盘，目前已经开发完成了windwos加速和linux加速。分别是两个工程。代码区别不大，主要是cmaklist.txt