# UNet

**Repository Path**: lllssskkk/UNet

## Basic Information

- **Project Name**: UNet
- **Description**: UNet 的套件模块

使用pytorch实现
- **Primary Language**: Python
- **License**: Unlicense
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 3
- **Forks**: 3
- **Created**: 2024-08-28
- **Last Updated**: 2025-08-12

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

# UNet

#### 介绍
UNet 医学影像切割

使用pytorch实现，其中model部分取自https://github.com/milesial/Pytorch-UNet

#### 安装教程

1.  [安装CUDA](https://developer.nvidia.com/cuda-downloads)
2.  创建虚拟环境(`venv`或者`conda`)
3.  [安装pytorch](https://pytorch.org/get-started/locally/) , 注意选择CUDA版本的pytorch安装
4.  安装依赖项`pip install -r requirements.txt`

#### 使用说明

1.  在configs.json文件中设置好数据的路径
2.  运行train.py 训练模型
3.  在predict.py 中调用训练好的模型来生成切割后的图片

#### 参与贡献

1.  Fork 本仓库
2.  新建 Feat_xxx 分支
3.  提交代码
4.  新建 Pull Request