自定义博客皮肤VIP专享

*博客头图:

格式为PNG、JPG,宽度*高度大于1920*100像素,不超过2MB,主视觉建议放在右侧,请参照线上博客头图

请上传大于1920*100像素的图片!

博客底图:

图片格式为PNG、JPG,不超过1MB,可上下左右平铺至整个背景

栏目图:

图片格式为PNG、JPG,图片宽度*高度为300*38像素,不超过0.5MB

主标题颜色:

RGB颜色,例如:#AFAFAF

Hover:

RGB颜色,例如:#AFAFAF

副标题颜色:

RGB颜色,例如:#AFAFAF

自定义博客皮肤

-+
  • 博客(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

空空如也

TA创建的收藏夹 TA关注的收藏夹

TA关注的人

提示
确定要删除当前文章?
取消 删除