深度学习工具链
关注
分享
扫一扫
文章平均质量分 71
主要记录一些深度学习工具链（如pytorch、fastreid、tensor RT等）相关的使用方法和注意事项。
TigerZ*
计算机视觉相关。接付费咨询开发。
算法：AIGC、分类、检测、特征、长尾、开集、聚类、传统图像算法。
语言：Python、C & C++。
工具：pytorch、TRT、CUDA、Triton、DeepStream、DALI。
展开
-
cv-cuda （cvcuda、nvcv）教程——Python安装
由于当前版本安装后，大家反应import nvcv cvcuda 失败，看官方文档，当前还不是很规范，特此记录当前版本的安装方法。原创 2023-01-11 17:07:42 · 4190 阅读 · 8 评论 -
视频分类——C3D使用
文章介绍了视频分类模型和开源代码的使用。原创 2024-05-21 18:38:40 · 942 阅读 · 0 评论 -
yolov5 优化——mosaic相关
Mosaic 利用了四张图片重新排布成一张图片，优点是丰富检测物体的背景、增加了很多小目标，让网络的鲁棒性更好。但是，如果我们的数据集本身就有很多的小目标，那么Mosaic数据增强会导致本来较小的目标变得更小，导致模型的泛化能力变差。如果我们的类目的关键信息是框的某个局部（某些场景你又不能标局部）那么裁切可能会将这个局部信息裁切没，导致模型无法拟合真正的信息。本文旨在如何修改mosaic来对特定场景进行调优。...原创 2022-08-17 19:52:29 · 4866 阅读 · 8 评论 -
yoloV5 教程——tensorboard使用
yolov5下tensorboard 的使用。原创 2023-02-02 20:01:43 · 4705 阅读 · 2 评论 -
YOLOV5 V6.1 详细训练方法
手把手教学如何使用YOLOV5训练检测模型原创 2022-07-13 20:05:22 · 1876 阅读 · 0 评论 -
yoloV5 使用——训练速度慢，加速训练
首先大家会按照官方的readme文件进行安装，连接：https://github.com/ultralytics/yolov5 此时有两点需要注意： 1、强烈建议大家使用conda虚拟环境安装，这样不会扰乱其他项目的Python环境，关于conda虚拟环境的使用，建议参考我的另一篇博客：https://blog.youkuaiyun.com/u012863603/article/details/115764553 举例如下：conda create --name y...原创 2021-08-17 15:38:08 · 16208 阅读 · 7 评论 -
fuser 使用—— YOLOV5内存溢出——kill nvidai-smi 无pid 的 GPU 进程
场景： 当我们使用YOLOV5训练模型时，官方的建议也是使用尽可能大的batch size这样BN更加的准确，训练速度也最快，但是当我们的batch size设置的过大会导致显卡内存溢出，而此时如果是多卡训练，很有可能出现长时间无法退出占用显存。解决方法： 第一种：使用nvidia-smi，下方有对应的pid。 可以使用kill命令杀掉进程，如下图，可以用如下命令： sudo kill -9 35346 # 杀掉pid为35346的进程 sudo ...原创 2021-08-18 14:46:22 · 3428 阅读 · 0 评论 -
GPU利用率低常见原因分析及优化方式
本文的 GPU 利用率主要指 GPU 在时间片上的利用率。GPU 任务会交替的使用 CPU 和 GPU 进行计算，当 CPU 计算成为瓶颈时，就会出现 GPU 等待的问题，GPU 空跑那利用率就低了。那么优化的方向就是缩短一切使用 CPU 计算环节的耗时。...转载 2022-08-17 10:35:13 · 15412 阅读 · 0 评论 -
mac m1 m2 深度学习环境（pytorch）配置
在mac M1 M2上配置Python 深度学习开发环境（pycharm、openCV、pytorch）原创 2022-06-23 11:52:04 · 11166 阅读 · 12 评论