U-NET安装与训练问题实录

normalone1

已于 2022-05-27 22:36:23 修改

阅读量992

点赞数 3

CC 4.0 BY-SA版权

分类专栏：深度学习文章标签： pytorch 人工智能 python

于 2022-05-27 22:22:13 首次发布

本文链接：https://blog.youkuaiyun.com/normalone1/article/details/125011016

本文详细记录了使用PyTorch安装U-Net模型的过程，包括从GitHub获取代码、环境配置、数据集验证、模型训练与预测，以及在训练和预测中遇到的问题及解决方案，特别指出输入数据格式和蒙版文件类型的重要性。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

参考文章：pytorch复现U-Net 及常见问题汇总（2021.11.14亲测可行）_奶盖芒果的博客-优快云博客_pytorch 无法复现

1.安装过程

1.1 github代码

代码网址：https://github.com/milesial/Pytorch-UNet

1.2 环境配置

requirements:
matplotlib
numpy
Pillow
torch
torchvision
tqdm
wandb

requirements中未包含具体版本信息，笔者亲测torch11.3 + python3.6可用

1.3 网络数据集验证模型可用

1.3.1 数据集地址

kaggle:Carvana Image Masking Challenge | Kaggle

数据集内容，这里我们下载tran.zi

最低0.47元/天解锁文章

200万优质内容无限畅学

确定要放弃本次机会？

福利倒计时

: :

立减 ¥

普通VIP年卡可用

立即使用

normalone1

关注关注

3
点赞
踩
7

收藏

觉得还不错? 一键收藏
0
评论
分享

复制链接

分享到 QQ

分享到新浪微博

扫一扫
举报

举报

专栏目录

如何安装U-net包以及跑HED的流程

BoJack_Horse的博客

08-09

1157

如何安装U-net包以及跑HED的流程安装anaconda：方法详见MOOC 安装tensorflow：试了很多次，tensorflow1.5.0版本能很合适安装U-net的包，同时也兼容HED的代码,但是按照上述方法装的anaconda里面只有1.1.0等较低版本，所以要这样下：注：由于不想下多了Jupyter，下这个老是出错不知道怎么回事，所以没有另外开一个环境专门装，直接...

【免费下载】 Pytorch-UNet安装与配置完全指南

最新发布

gitblog_09465的博客

09-13

2130

Pytorch-UNet安装与配置完全指南项目基础介绍及编程语言项目名称： Pytorch-UNet 主编程语言： Python 关键依赖： PyTorch 目的：此项目是PyTorch实现的U-Net模型，专为图像语义分割设计，特别是在高分辨率图像上的应用，如Kaggle的Carvana图像掩模挑战赛。它提供了一种高效的方法来训练和预测图像中的目标区域，得分高，精度优秀。关键技术和框架 ...

参与评论您还未登录，请先登录后发表或查看评论

第一次运行U-Net遇到的问题

qq_41772084的博客

03-13

440

用自己笔记本跑的 epoch21的时候出错，有篇博客分析原因是内存溢出，我这的提示描述的和人家不一样我也不知道什么原因，也把30改成了20，然后训练，预测全灰，网上搜索说是因为样本少、迭代次数少。我迭代次数确实少，增加迭代次数试试。 ...

安装nnUNet

weixin_49178809的博客

08-02

1067

在D盘里直接新建一个文件夹命名为nnUNetFrame，然后在文件夹内cmd打开git clone https://github.com/MIC-DKFZ/nnUNet.git。会出现内存不够的报错，numpy.core._exceptions._ArrayMemoryError然后更改了虚拟页面大小，减少数据集。因为直接安装了python不想卸载了就没装anaconda，直接搜不用安装anaconda安装pytorch即可。这一步报错了，查了一下因为开了VPN端口被占用了，关了就行。

U-Net 开源项目安装与使用指南

gitblog_00514的博客

08-20

425

U-Net 开源项目安装与使用指南 1. 目录结构及介绍在深入到U-Net项目的细节之前，首先了解一下其基本的目录结构。请注意，我并未直接访问提供的GitHub链接以获取实时信息，但通常一个基于U-Net的项目可能会有以下典型结构： u-net/ ├── README.md # 项目说明文件，包含了快速入门指导和重要信息。 ├── requirements.txt # 项...

《守望先锋》架构设计与网络同步 -- GDC2017 精品分享实录.rar

09-16

4. **服务端验证**：考虑到作弊问题，游戏可能会采用服务端验证机制，即客户端发送的动作需要经过服务器确认后才能生效，保证了游戏的公平性。接下来，网络同步是在线游戏中至关重要的技术，尤其是在快节奏的射击...

大数据教程-Sqoop安装与使用实录.pdf

07-26

根据给定的文件信息，以下是关于大数据教程中Sqoop安装与使用实录的知识点： ### 大数据教程-Sqoop安装与使用实录 #### 1. Sqoop简介 Sqoop是一个开源工具，主要用于在Hadoop与传统数据库（如MySQL、Oracle等）...

联想G40-70m黑苹果安装实录(EFI备份）

04-19

由于本人机型设备（显卡）和GitHub上的不一致，自己重新配置了一遍，在此作为备份，有需要的朋友可以下载使用。其中包含了window系统和Ubuntu18.04系统，都是通过clover引导启动的，虽然这样对于后续的升级有影响，...

变化--1990-2002年中国实录

12-26

这是一本描述变化的书。 1990－2002年里，我们每一个人都在不知不觉中经历了这种变化；今后，我们还将经历更大的变化。在作者笔下，1990－2002年中国的格外引人入胜之处，不是在于她的轰轰烈烈，而是在于她的平淡...

U-Net

weixin_41874898的博客

08-28

290

U-Net 1 安装libtiff 问题：pip install libtiff没有办法直接安装解决：手动安装，去https://www.lfd.uci.edu/~gohlke/pythonlibs/#pygame下载libtiff，转到对应下载的文件夹目录 pip install libtiff-0.4.2-cp36-cp36m-win_amd64.whl 2 rindex 返回子字符串 ...

tf_unet, 用于图像分割的通用U 网 Tensorflow.zip

09-17

tf_unet, 用于图像分割的通用U 网 Tensorflow Tensorflow Unet 这是由 Ronneberger等提出的通用英镑实现。开发了 Tensorflow 。方法已经应用于射频干扰抑制，并利用深卷积神经网络( ) 进行了抑制。网络可以训练在任意成像数据上

u-net 模型训练

雪剑封心

01-27

430

使用数据增强 imgaug库。

Unet

光明〇博客

10-28

597

下载Unet资源后，包内有.py和.ipynb文件，在Spyder（TensorFlow）环境中依次运行data.py、model.py、main.py文件，过程中会报错：少keras，安装pip install kernel ，；少skimage，安装pip install scikit-image；少pillow，安装pip install pillow。等训练结果 ...

【nnUNet】环境安装

qq_40041064的博客

09-04

671

本文分享医疗分割模型nnUNet的环境安装教程，希望对各位小伙伴有所帮助！1. 安装Python & PytorchPIP安装Pytorch_pip pytorch]-优快云博客安装完成后，检查Pytorch的GPU版本是否可用。在输出为true时，表示可用。2. 安装nnUNet在nnUNet文件目录，运行Cmd命令安装安装成功3. 设置环境变量使用时替换为自己的路径。

U-net+代码理解+数据流训练法1

weixin_44886683的博客

12-23

1789

网络结构图 Unet前世传奇什么是图像分割问题呢？简单的来讲就是给一张图像，检测是用框出框出物体，而图像分割分出一个物体的准确轮廓。也这样考虑，给出一张图像Ｉ，这个问题就是求一个函数，从I映射到Mask。至于怎么求这个函数有多种方法。我们可以看到这个图，左边是给出图像，可以看到人和摩托车，右边是分割结果。求这个函数有很多方法，但是第一次将深度学习结合起来的是这篇文章全卷积网络(FCN)，...

Windows Pychram部署Unet++过程

ArcGis_Niu的博客

09-21

608

windows系统在pycharm平台上配置并跑通Unet网络这里我用的是B站课程里的资源《唐宇迪Unet图像分割语义分割实战》

nnU-Net 如何安装--傻瓜式安装教程

Tina姐的博客

05-25

9566

这是nnU-Net系列第二篇–安装教程上一篇：nnU-Net介绍篇 nnU-Net只在Linux上测试（ubuntu16、18和20；centOS, RHEL）。它也可以在其他操作系统上工作，但我们不能保证它会工作。因此，最好安装在Linux上文章目录安装nnU-Net的硬件条件快速的创建 Ubuntu 虚拟环境1 安装venv模块2 创建虚拟环境存放目录3 创建虚拟环境4 激活该环境5 退出该环境安装nnunet设置环境变量如何设置环境变量设置路径的另一种方法注意安装nnU-Net的硬件条件 nn

学习笔记(01):TensorFlow 2.0 三大项目实战-训练U-NET模型

qq_35040443的博客

03-12

712

TensorFlow是由Google开发的深度学习框架，被广泛应用于各类机器学习算法的编程实现，无论你是专家还是初学者，它都可以帮助你轻松构建和部署机器学习模型。作为当下最为热门的机器学习框架，TensorFlow具有非常完整的生态系统，可以帮助你使用机器学习解决各种棘手的现实问题，并随时随地进行可靠的机器学习生产，英特尔、AIRBUS、中国移动、小米等一大批知名企业都在使用Tenso...

U-2-Net

jacke121的专栏

10-29

2931

https://github.com/NathanUA/U-2-Net