# ExitU-Net

**Repository Path**: sdu-aes-lab/exit-u-net

## Basic Information

- **Project Name**: ExitU-Net
- **Description**: 面向嵌入式设备的高效图像分割系统
- **Primary Language**: Unknown
- **License**: Not specified
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 0
- **Forks**: 0
- **Created**: 2025-05-08
- **Last Updated**: 2025-07-29

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

# 面向嵌入式设备的高效图像分割系统

## 使用说明
### 环境配置
1.  安装CUDA和1.13版本及以上的PyTorch
2.  安装各种依赖：

```
pip install -r requirements.txt
```
### 模型训练
1.  将数据集的图片和分割标注分别放在./data/imgs/和./data/msks/下
2.  训练模型：
```
python train.py
```
3.训练好的模型将会保存到./checkpoints/下
### 模型的使用和测试
1.  使用模型分割图片：
```
python predict.py -i image.jpg -m MODEL.pth
```
2.  计算模型的dice系数和退出层级：
```
python test.py
```
3.  计算模型参数量和计算量
```
python ./params/params.py
```
4.  计算模型推理速度
```
python FPS.py
```
```
python FPSunet.py
```