0. 本文内容

  1. Github 上U-Net原数据集跑通
  2. 环境配置(pytorch[cpu,gpu])
  3. 自己数据集复现
  4. 其他

1. 原数据集跑通

github地址
下载程序。下载下来不包含数据集
从其他博主捞到的含数据集的,提取码:oi6e ,包含一个汽车数据集和细胞数据集。程序是GitHub下载下来的程序。

运行直接运行train.py文件即可。

2. 环境配置

anaconda,pytorch-gpu版本。

  1. 安装anaconda,创建环境。
    在这里插入图片描述

  2. 打开prompt,激活环境。
    在这里插入图片描述activate unet1
    在这里插入图片描述

  3. 配置pytorch
    在这里插入图片描述--------按照红色区域选择,复制红色框的内容到prompt运行。(其中第一行是torch版本,第三行是安装方式,conda安装比较方便。最后一行是选择要配置的CUDN的版本)
    --------等待下载完毕。如果运行超时,给conda换个下载路径,静静等待安装完毕。
    --------测试是否安装成功:import torch(尝试调用这个库)

  4. 安装其他需要的库

#比较快的方式
pip install --user 库名  -i https://pypi.tuna.tsinghua.edu.cn/simple

3. 自己数据集复现

分享的文件里包含了两个数据集,一个是car,可以直接运行的,一个是细胞的,从博主淘来的。
眼球毛细血管数据集
自己下载到本地,在程序里改调用地址就可以。

更改数据集需要修改的参数:

  1. 调用地址(改成自己的)
    在这里插入图片描述

  2. 图片格式
    查看imgs与masks的图片格式,修改对应的后缀。
    在这里插入图片描述

  3. 修改读取mask的名称格式。
    在这里插入图片描述
    masks的图片前半部分和imgs的一致,多出来的字符串添加到引号内,完全一致就为空。

不出问题的话到此可以正常运行自己的数据集了。

如果需要自己制作数据集,有labelme这个软件,自己搜教程。

4. 其他

程序开始的时候会显示使用的是cpu还是cuda
在这里插入图片描述

Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐