- 博客(4)
- 收藏
- 关注
RSS订阅原创 使用segmentation model pytorch训练自己的数据集
如果要使用UNet、FPN、UnetPlusPlus等模型，只要改成model = smp.Unet（） or model = smp.FPN（） or model = smp.UnetPlusPlus()即可。包含背景在内，分割类别共4类: ["bg", "class1"，"class2", "class3"]数据集主目录中包含以下几个文件夹，分别是训练集的原图和mask 、测试集的原图和mask。在mask中的像素值分别为0，64，128，192。训练的时候还遇到了问题，显示报错。
2025-03-03 11:45:55
726
2
原创 segmentation model pytorch安装记录
是一个基于PyTorch的库，旨在简化语义分割模型的使用和训练过程。它提供了多种流行的架构和预训练权重，能够快速地进行实验、开发和部署语义分割任务。
2025-02-26 17:48:17
805
原创 深度学习停止训练后，显存仍然占用的问题
停止深度学习训练后，使用nvidia-smi查看 发现显存仍然被占用。输入 ps aux | grep python。再次输入nvidia-smi查看显存占用情况。输入 kill -9 PID。找到对应进程的PID。
2025-02-21 15:06:02
221
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人