随笔分类 - 深度学习
摘要：netron查看模型，进行区分 // 从TensorRT输出维度 动态获取 分类数int FormTensorRT::get_num_classes(Dims output_dims) { // YOLO输出格式: [1, na*(5 + nc), 8400] 或 [1, 8400, na*(5+n
阅读全文
摘要：bool build_ok = buildAndSaveEngine(ONNX_PATH, TRT_ENGINE_PATH); if (!build_ok) { ShowMsg("ONNX 转 TRT 失败！"); delete runtime; return; } 1.显卡需要rtx3060 2.
阅读全文
摘要：labelimg打不开，已安装，报错。卸载-》重新安装 anaconda激活pytorch环境 pip uninstall labelimg pip install labelimg
阅读全文
摘要：Traceback (most recent call last): File "D:\ttt\yolov5-master\train.py", line 773, in <module> main(opt) File "D:\ttt\yolov5-master\train.py", line 56
阅读全文
摘要：engine = runtime->deserializeCudaEngine(engineData.data(), engineData.size()); if (!engine) { ShowMsg("引擎创建失败"); delete runtime; return; }
阅读全文
摘要：trtexec --loadEngine=model.trtrtx3060显卡显示pass则OK &&&& PASSED TensorRT.trtexec [TensorRT v100001] # trtexec --loadEngine=model.trt D:\cdtxw\TensorRT-10
阅读全文
摘要：TensorRT推理Onnx #include "FormTensorRT.h" #include <QToolButton> #include <QMessageBox> #include <QDebug> #include <iostream> #include <fstream> #inclu
阅读全文
摘要：https://developer.nvidia.com/tensorrt/download/10x https://developer.nvidia.com/downloads/compute/machine-learning/tensorrt/10.0.1/zip/TensorRT-10.0.1
阅读全文
摘要：gpu推理报错 缺少：cudnn_heuristic64_9.dll Non-zero status code returned while running Conv node. Name:'Conv_0' Status Message: Failed to initialize CUDNN Fro
阅读全文
摘要：这个版本onnxruntime-win-x64-gpu-1.24.2 对应cuda_12.4.1，cudnn9.21 https://txwtech.blog.csdn.net/article/details/160433097 https://developer.nvidia.com/cuda-t
阅读全文
摘要：auto outputTensors = session.Run(Ort::RunOptions{ nullptr }, inputNames.data(), &inputTensor, 1, outputNames.data(), 1); yolov5推理Onnx出现 inputNames字符含有
阅读全文
摘要：OMP: Error #15: Initializing libiomp5md.dll, but found libiomp5md.dll already initialized. 在你的训练脚本（例如 train.py）的最开头（即在导入 torch、numpy 或其他任何库之前）添加以下两行代码
阅读全文
摘要：PS D:\deep_learning\yolov5-master> D:\anaconda3\envs\pytorch\python.exe D:/deep_learning/yolov5-master/detect.py --weights side_defect.onnx --source D
阅读全文
摘要：conda python如何查看onnx版本 (base) C:\Users\zzy>activate pytorch (pytorch) C:\Users\zzy>conda list onnx # packages in environment at D:\anaconda_install\en
阅读全文
摘要：yolov5训练与验证的指令 确认使用anaconda\envs下面的python.exe运行脚本 D:\anaconda_install\envs\pytorch\python.exe D:\zzy\yolov5-master\train.py --weights yolov5s1.pt --da
阅读全文
摘要：yolov5训练效果好，在识别时识别率低不好是什么原因？ 确认使用了正确的，最新训练完成的best.pt文件 放在yolov5-master目录目录下，与train.py和detect.py是相同目录
阅读全文
摘要：检索增强生成（Retrieval-augmented Generation），简称RAG
阅读全文
摘要：参考：https://www.cnblogs.com/guojin-blogs/p/18108220 onnx转tensorRT.engine 1. 项目介绍 NVIDIA® TensorRT™ 是一款用于高性能深度学习推理的 SDK，包括深度学习推理优化器和运行时，可为推理应用程序提供低延迟和高吞
阅读全文
摘要：各个版本介绍 https://blog.csdn.net/sinat_34897952/article/details/148014377 yolov5版本 https://github.com/ultralytics/yolov5 yolov8版本 https://github.com/ultra
阅读全文
摘要：参考：https://www.cnblogs.com/answerThe/p/15616811.html conf_thres Confidence Threshold，置信度阈值。 只显示预测概率超过conf_thres的预测结果。 想让YOLO只标记可能性高的地方，就把这个参数提高。 iou_t
阅读全文
浙公网安备 33010602011771号