- 博客(2)
- 收藏
- 关注
原创 手把手完成多分类图像分割----基于segmentation_models_pytorch框架
由于最近在进行医学图像的分割任务,写篇博客记录一下怎么从无到有完成一个多分类的医学图像分割。博客模型载入方面使用的是segmentation_models_pytorch框架,如果是在框架中没有的分割模型,也可以直接找论文model仓库copy项目主题框架如图所示data文件夹,用于放置图像数据以及mask文件weight文件夹,用于保存模型权重mydataset文件,mydataset文件分为两个,分别用来读取图像格式以及json格式的masktrain文件,主训练代码。
2023-08-21 10:45:20 1202 5
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人