- 博客(3)
- 收藏
- 关注
原创 BEFUNet训练注意事项
4.在训练中断时,我可以保存已经训练好的模型作为预训练模型继续进行训练(因为模型默认的预训练模型是Swin和pidinet模型)。在/root/autodl-tmp/BEFUnet-main/train.py文件中做如下修改。3.由于在运行train.py时,会隔每个eval_interval保存模型并进行推理,我嫌麻烦就把推理给省去,直接保存模型。2.修改BEFUnet-main/test.py第45行代码,添加default=True,要不然预测图不会被保存至指定文件夹。
2024-09-17 15:25:05
454
2
原创 nnUNetv2训练注意事项
2.在训练的过程中会出现RuntimeError: One or more background workers are no longer alive. Exiting. Please check the print statements above for the actual error message报错,一般是由于磁盘空间不足,扩容即可。3. 我的原始数据是png格式,在训练二维图像之前,由于这是2D图片数据,要将其转为3D数据,其实就是z轴为1的3维数据。
2024-09-15 20:41:34
1088
1
原创 MMSegmentation-Segformer跑自己的数据进行二分类
比如我的数’background’,‘building’在单通道的label数据中要像素值分别要为‘background’=0,‘building’= 1,因为我的数据是background’=0,‘building’= 255,所以我修改了源代码将标签中为255的像素值改成了1。2.遇到的评价指标为0或者100%的问题。我的标签只有两个类别。
2024-09-15 13:00:55
449
4
空空如也
添加商品信息为什么运行不出来?
2023-12-19
用void进行类型转换
2023-12-05
TA创建的收藏夹 TA关注的收藏夹
TA关注的人