- 博客(14)
- 资源 (2)
- 收藏
- 关注
RSS订阅原创 tensorRT安装
我下的版本是TensorRT-10.9.0.34.Windows.win10.cuda-11.8。4、将lib目录和include目录中的文件内容都拷贝到cuda对应的目录文件下。pycuda的版本根据需要选择。我的CUDA 是11.8，所以选的是2023.1。5、在python文件夹中有对应的Python安装库。有可能是pycuda的版本与电脑状的CUDA不对应。6、在samples中有相关的使用示例。2、根据系统和CUDA版本选择选一个。1、根据需要选择对应的版本。pycuda安转的那些事。
2025-04-03 14:57:01
221
原创 深度学习模型单任务该多任务
最近因为数据限制和任务要求的改变，需要将现有分类模型加一个分类头。但原有的分类数据集没有，只有权重。模型网络用的是timm里的tresnet_m。
2025-03-24 14:30:39
196
原创 onnxruntime安装
在安装onnxruntime时，如果直接运行“pip install onnxruntime-gpu”安装的是最新版的onnxruntime-gpu，且对应的是的（CUDA 12.x）。这样很有可能导致CUDA版本不对应导致GPU用不了。
2024-11-04 17:13:44
1354
原创 YOLO中为什么要freeze dfl.conv.weight‘
在训练YOLOv8代码的时候发现，无论怎么设置参数，都会有‘Freezing layer 'model.20.dfl.conv.weight’。意思就是这层的权重被冻结了。
2024-10-11 14:17:20
1277
3
原创 python图像保存
最近要保存标签文件时发现将标签文件保存成JPG格式会将原有数据发生变化，图像质量调到最高也是。查阅材料后发现，JPG格式格式下，即使将质量调到最高，也会将数据进行压缩处理，进而使其与原数据不一致。如下图所示：使用JPG格式保存后，图像出现噪点。这对于图像分割是存在不利影响的。所以在保存图像标签文件应该采用bmp、无压缩的npg、无压缩的tiff等。
2024-08-22 12:42:21
1252
原创 Albumentations中一些函数的变化
在使用albumentations提示有一些功能没法用，查看官方网文档时发现新版的一些功能的名字有所变化。AttributeError: module 'albumentations' has no attribute 'IAAAdditiveGaussianNoise'
2024-08-08 19:41:31
1239
1
原创 FastSAM分割一切
Segment Anything Model （SAM）论文翻译。它被认为是一个里程碑式的愿景基础模型。它可以在各种可能的用户交互提示的引导下分割图像中的任何对象。SAM 利用在广泛的 SA-1B 数据集上训练的 Transformer 模型，使其能够巧妙地处理各种场景和对象。SAM 为一项激动人心的新任务打开了大门，称为 Segment Anything。由于其可推广性和潜力，这项任务具有成为未来广泛视觉任务的基石的所有条件。
2024-08-07 16:58:35
1780
原创 torchio.transforms
随机仿射变换，包括尺度（scale，需要指定缩放的比例，可以设置缩放时的差值策略），旋转（degrees，需要指定每个轴旋转的角度范围，可以设置旋转时pad的数值），平移（translation），还支持各向同性和设置以中心为基准进行变换。沿三个轴缩放，每个轴的缩放范围为（1-x_i, 1+x_i）沿三个轴旋转，每个轴的旋转范围为（-x_i, x_i）沿三个轴平移，每个轴的平移范围为（-x_i, x_i）沿三个轴旋转，每个轴的旋转范围为（a_i, b_i）沿第一个轴向旋转，旋转范围为（-x, x）
2022-09-12 17:13:19
609
原创 图像张量分块
tensor的分块与还原代码代码from PIL import Imageimport torchvision.transforms as transformsimport matplotlib.pyplot as pltimport SimpleITK as stkimport torch# 读取图像并转为tensordef readImage(path='D:\微信图片_20200304231857.jpg', size=1000):#这里可以替换成自己的图片 mode = Im
2021-08-25 20:41:59
294
原创 Python数据读取处理与导出
经过多方查找拼接了一个将Excel中的数据导入到Python中、数据处理、将处理后的数据到出到Excel中的整个程序，并做了标注import pandas as pdimport numpy as npdat = pd.read_excel('C:/Users/Administrator/Desktop/数据/qw.xlsx') #读取Excel中的数据ded=list(dat['...
2018-07-29 17:56:56
6615
新版ffmpeg-2024-12-26-git-fe04b93afa-full-build.7z
2024-12-30
Labelme AI模型权重
2024-08-30
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人