U-Net++细胞分类

1.问题描述

引用albumentations.augmentations.transforms时未找到module：RandomRotate90
在做Unet 细胞分类的时候。导入albumentations.augmentations.transforms时未找到module：RandomRotate90，和Resize

2.解决办法

在导入文件库里面加上

import albumentations as A

把出错的函数改为：

A.RandomRotate90  
A.Resize

在这里插入图片描述

3.代码解析

源码地址：https://gitee.com/braveeeeeeeeeeeeeeeeeee/segnentation
在这里插入图片描述
train.py先训练数据集得到权重矩阵模型：.pth文件
train.py参数配置：
在这里插入图片描述

在这里插入图片描述
val.py然后就可以用.pth权重参数在验证模块测试：
val.py里面需要加载一下.pth文件夹位置：
在这里插入图片描述
在这里插入图片描述

4.原理介绍

在这里插入图片描述

在这里插入图片描述

在这里插入图片描述

5.结果展示

image.png

版权声明：本文为weixin_43694742原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接和本声明。