molyswu
commented
4 months ago 使用 trtexec 转换模型
trtexec --onnx=model.onnx --saveEngine=model.engine
代码：
int main() {
// patchcore模型训练配置文件调整center_crop为 center_crop: null
// trtexec --onnx=model.onnx --saveEngine=model.engine 转换模型
string model_path = "D:/2011/anomalib/results/patchcore/tube/run/weights/onnx/model.engine";//"D:/ml/code/anomalib/results/efficient_ad/mvtec/bottle/run/weights/openvino/model.engine";
string meta_path = "D:/2011/anomalib/results/patchcore/tube/run/weights/onnx/metadata.json";
string image_path = "D:/2011/anomalib/datasets/D3.png";
string image_dir = "D:/2011/anomalib/datasets/img_192/abnormal/abnormal";
string save_dir = "D:/2011/python_work/NagatoYuki093/anomalib-tensorrt-cpp-main/anomalib-tensorrt-cpp/result"; // 注意目录不会自动创建,要手动创建才会保存
bool efficient_ad = true; // 是否使用efficient_ad模型
// 创建推理器
auto inference = Inference(model_path, meta_path, efficient_ad);
// 单张图片推理
cv::Mat image = readImage(image_path);
Result result = inference.single(image);
saveScoreAndImages(result.score, result.anomaly_map, image_path, save_dir);
cv::resize(result.anomaly_map, result.anomaly_map, { 1500, 500 });
cv::imshow("result", result.anomaly_map);
cv::waitKey(0);
// 多张图片推理
inference.multi(image_dir, save_dir);
return 0;
}
NagatoYuki0943
谢谢！
我的意思是像mvtec上图的第5个图会画出来。
将你的工程改为VS2022 加载model.engine模型，出现溢出， Expression: vector subscript out of range for information on how your program can cause an asserti。
“anomalib-tensorrt-cpp.exe”(Win32): 已加载“C:\Windows\System32\windows.storage.dll”。 “anomalib-tensorrt-cpp.exe”(Win32): 已加载“C:\Windows\System32\SogouPY.ime”。 “anomalib-tensorrt-cpp.exe”(Win32): 已加载“C:\Windows\System32\oleacc.dll”。 “anomalib-tensorrt-cpp.exe”(Win32): 已加载“C:\Windows\System32\winhttp.dll”。 “anomalib-tensorrt-cpp.exe”(Win32): 已加载“C:\Windows\System32\ntmarta.dll”。 “anomalib-tensorrt-cpp.exe”(Win32): 已加载“C:\Windows\System32\profapi.dll”。 “anomalib-tensorrt-cpp.exe”(Win32): 已加载“C:\Program Files (x86)\SogouInput\14.2.0.8968\Resource.dll”。 “anomalib-tensorrt-cpp.exe”(Win32): 已加载“C:\Program Files (x86)\SogouInput\Components\PicFace\1.1.0.2129\PicFace64.dll”。 “anomalib-tensorrt-cpp.exe”(Win32): 已加载“C:\Windows\System32\psapi.dll”。 “anomalib-tensorrt-cpp.exe”(Win32): 已加载“C:\Windows\System32\dwmapi.dll”。 线程 13220 已退出，返回值为 0 (0x0)。 “anomalib-tensorrt-cpp.exe”(Win32): 已卸载“C:\Windows\System32\SogouTSF.ime” 0x00007FF818C8829C (ucrtbased.dll) (anomalib-tensorrt-cpp.exe 中)处有未经处理的异常: 将一个无效参数传递给了将无效参数视为严重错误的函数。
请教如何处理？
谢谢！ molyswu