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.结果展示