2026年5月27日
从TensorRT输出维度 动态获取 分类数onnx转trt
摘要： netron查看模型，进行区分 // 从TensorRT输出维度 动态获取 分类数int FormTensorRT::get_num_classes(Dims output_dims) { // YOLO输出格式: [1, na*(5 + nc), 8400] 或 [1, 8400, na*(5+n 阅读全文
posted @ 2026-05-27 21:53 txwtech 阅读(20) 评论(0) 推荐(0)
ONNX 转 TRT 失败-tensorRT
摘要： bool build_ok = buildAndSaveEngine(ONNX_PATH, TRT_ENGINE_PATH); if (!build_ok) { ShowMsg("ONNX 转 TRT 失败！"); delete runtime; return; } 1.显卡需要rtx3060 2. 阅读全文
posted @ 2026-05-27 14:59 txwtech 阅读(7) 评论(0) 推荐(0)
labelimg打不开，已安装，报错。卸载-》重新安装
摘要： labelimg打不开，已安装，报错。卸载-》重新安装 anaconda激活pytorch环境 pip uninstall labelimg pip install labelimg 阅读全文
posted @ 2026-05-27 09:52 txwtech 阅读(17) 评论(0) 推荐(0)