- 博客(5)
- 收藏
- 关注
RSS订阅原创 TransUnet官方代码训练自己虹膜分割数据集（错误解决）
压成npz文件的时候先看看自己的label标签是什么，如果是如下图和我一样二分类，我的类别是0和255，即黑（0）白（255）色，就要在压成npz的时候改成0，1。类别多的同理改成0开始的连续数字，下面是我的制作npz代码。如果输入的图片不是正方形就写最长边，例如我是320*280，设置 img_size = 320。开始我先把我的图像压成npz格式后导入了这两个文件夹，下面仔细检查train.py和test.py文件，我用的CSASIA-V4数据集，用什么都一样。运行train.py,运行成功。
2024-11-29 09:58:57
544
3
原创 ResNet学习笔记
resnet分为resnet18、resnet34、resnet50、resnet101、resnet152把他们分为两种，其中 resnet18、resnet34 是一组；resnet50、resnet101、resnet152 是一组。两个不同的组分别使用下面两种不同的残差块（ResNet block），分别为两层的结构和三层的结构。其中深层次时故50往上的结构都用的 block。
2024-10-17 17:17:43
717
原创 UserWarning: CUDA initialization: CUDA driver initialization failed, you might not have a CUDA gpu.
在安装pytorch最后的验证步骤时输入import torch，提示UserWarning: CUDA initialization: CUDA driver initialization failed, you might not have a CUDA gpu.不匹配，注意：可以输入nvidia-smi和nvcc -V查看自己cuda版本，其中前者的要大于后者的才行。在任务管理器-性能-GPU1看自己的GPU，进官网对比下载最新的版本，一直点下一步。后面看了上面那篇博客才解决问题。
2024-09-16 16:26:50
719
1
空空如也
swin transformer
2024-10-19
pycharm打开labelme出现
2024-06-13
TA创建的收藏夹 TA关注的收藏夹
TA关注的人