Windows Pychram部署Unet++过程

*针对完全0基础的萌新而写~各位大牛见笑了 *

1 下载Unet++包

这里我用的是B站课程里的资源,课程名称《人工智能必备【图像分割+语义分割】经典项目实战,计算机大牛手把手教你做unet医学细胞分割实战,草履虫都说简单!》
资料专题为:《唐宇迪Unet图像分割语义分割实战》
资料链接::https://pan.baidu.com/s/1D7bvk2vco4vexCH92Iz8PQ?pwd=860r
提取码:860r
课程截图

2 环境配置

2.1 虚拟环境

笔者用的是windows系统,编译器为pycharm,在笔者自己创建的虚拟环境中运行。虚拟环境python版本是3.9.16
在这里插入图片描述

2.2 pycharm中相关库的安装

鼠标悬停,在有红色波浪线的库下面选择安装,重复操作直到都安装好
在这里插入图片描述

2.3 对数据集进行预处理

修改preprocess_dsb2018,第15行,改成自己想要的数据预处理的路径。
这里的预处理主要是将\unet++\inputs\stage1_train\00ae65c1c6631ae6f2be1a449902976e6eb8483bf6b0740d00530220832c6d3e\masks路径中的所有单个mask合并到一张mask图上,
在这里插入图片描述

在这里插入图片描述
在这里插入图片描述

2.4 运行train.py的debug过程

进行了预处理之后就可以进行训练了,运行train.py文件

3 题外话 debug过程记录

3.1 AssertionError: Torch not compiled with CUDA enabled

这个错误是由于没有CUDA造成的。笔者的环境是cpu,没有下载cuda,也没有gpu。所以这里就需要对代码中调用了CUDA的部分都进行一个替换,改成cpu运行。

1、先在代码头部加上这一行代码(注意区分中英文符号

 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")

在这里插入图片描述

2、然后将代码中所有 .cuda() 替换为 .to(device)
就可以在无cuda的环境中运行了
在这里插入图片描述

3.2 module ‘albumentations.augmentations.transforms’ has no attribute ‘Resize’

这是由于自动安装的albumentations库版本太高,不兼容。只需要在File——Settings——Python Interpreter中将心啊在安装的albumentations卸载,然后安装1.1.0版本即可
先卸载现存albumentations
安装1.1.0版本的albumentations库
在这里插入图片描述

3.3 其他报错

AttributeError: module ‘albumentations.augmentations.transforms’ has no attribute ‘RandomRotate90’
AttributeError: module ‘albumentations.augmentations.transforms’ has no attribute ‘Flip’
AttributeError: module ‘albumentations.augmentations.transforms’ has no attribute ‘Resize’

解决方法:
首先

import albumentations as albu

然后将报错的这几个属性都替换为albu.
在这里插入图片描述

猜你喜欢

转载自blog.csdn.net/ArcGis_Niu/article/details/132218610