- 博客(23)
- 收藏
- 关注
RSS订阅原创 pytorch中，如何灵活使用dataloader
因为项目需求，我现在项目中有个需求：现在我有两个yolo数据集，我需要各取两个数据集中1：1然后同一个batchsize，我是打算防止写出两个yaml，将两个数据合并在一个yolo格式的yaml，然后一个数据集为1_00000.png,另一个数据集以0_000001命名，然后关于dataloader的源码，首先我们看看这部分的yolo源码在/home/san/ai_code/ai/ultralytics-main/ultralytics/engine/trainer.py中290行左右点进去是。
2025-05-20 16:18:45
1128
1
原创 docker清洁内存指南
我们在使用docker的时候总会遇到commit多次，这会产生很多没必要的内存使用这行代码可以列出现在容器的内存情况运行的结果为 我们来解释下这些的意义也就是你用看到的镜像，包括：你下来的镜像（如你自己构建的镜像被容器使用或未使用的镜像✅ 活动镜像（ACTIVE）：正在被容器引用，或有标签（tag）的镜像✅ 可清理空间（RECLAIMABLE）：没有 tag 的镜像（如<none>已经没被任何容器使用的旧版本镜像可以通过清理包括：正在运行的容器（docker ps。
2025-04-29 15:07:43
433
原创 rknn推理从入坑到入土
rknn是瑞芯微的一个npu推理框架，相对于老黄的生态，虽然生态没有cuda生态好，但是起码坑比较少，以下总结下如何开始推理，以yolov8为例。
2024-12-11 11:15:17
2099
1
原创 如何在python调用cpp的dll或者so库
方法1：直接把输出的pyd和需要调用的dll（或者加环境变量）库放在python main函数同一文件夹下（linux直接放so文件就行），然后python直接import就行，如果编译器爆红也请大胆尝试跑，因为是正常的现象。本文使用的是pybind11来调用so和dll（pyd），实现0时间损失调用，以下会用yolov11 pose的tensorrt部署来举例。因为为了满足一些框架无法调用cpp的目的，折腾了几天还是成功搞出来了，而且这个方法目前测试基本和cpp运行时间，有什么问题欢迎评论区。
2024-11-22 13:59:15
477
原创 从0开始搭建linux镜像docker深度学习tensorrt部署环境
保存的文件默认在 C:\Users\13951>命令行所在的文件。1.在windows拉取docker镜像（可省略）安装docker这里就不再阐述了，网上一搜一大把。具体需要什么版本的tensor rt要怎么改。2.在Linux上加载docker镜像。docker file可以这样写。然后在将拉下来的英伟达进行保存。3.配置tensorRT环境。4.进入镜像配置其他环境。
2024-11-12 14:26:07
297
原创 docker自用配置tensorRT&CLion环境教程
当你拉下来一个环境后，我们现在如何开始配置呢1：查看目前所有的容器2：删除容器强制删除运行容器3：运行并且进入容器。
2024-10-31 14:06:33
449
1
原创 openvino推理保姆教学
二：推理（分割模型举例 （N,C,H,W）-> (N,C,H,W)）就是那么简单，没什么技术含量，主打一个调api就行。然后就输出指针你想怎么折腾就怎么折腾去吧。
2024-07-22 17:10:37
320
原创 cmake相关学习记录
定义 OpenCV 库列表，将在链接时使用。：递归搜索并收集项目中的所有源文件和头文件。1.创造cmake要求和项目名称。以DLL_API来定义API接口。4.设置 OpenCV 库列表。3，包含项目的头文件和库目录。
2024-07-11 13:31:12
331
原创 Qt从入门到入土（第一天）
（第二步）安装vs插件，一样在官网，找不到的话就直接vs插件里面找。（第三步）添加到vs配置和环境变量。vs stdio不多讲。
2024-04-16 16:40:59
226
1
原创 深度学习中，动态改变batchsize（自用）
可以看出其实对于batchsize，有BatchSampler这个类，能写得非常非常灵活的。
2024-02-29 01:09:25
1193
1
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人