2023-04-07 22:32:17 +08:00
|
|
|
|
# 使用Pytorch简单实现SegNext，学习用
|
2023-04-07 22:40:24 +08:00
|
|
|
|
pretrained文件夹内放置权重文件
|
2023-04-07 22:32:17 +08:00
|
|
|
|
可以使用官方版本的tiny权重文件进行训练或者预测（其他版本未尝试）
|
|
|
|
|
|
更改运行参数请使用config文件夹内的yaml文件配置
|
|
|
|
|
|
训练移步train.py，预测移步predict.py, 有不会传的参数请移步main.py
|
2024-10-15 19:10:12 +08:00
|
|
|
|
|
|
|
|
|
|
文件应当按照如下形式组织
|
|
|
|
|
|
|
|
|
|
|
|
```
|
|
|
|
|
|
.
|
|
|
|
|
|
├── arguments.py
|
|
|
|
|
|
├── bricks.py
|
|
|
|
|
|
├── config
|
|
|
|
|
|
│ ├── lr_scheduler.yaml
|
|
|
|
|
|
│ ├── model.yaml
|
|
|
|
|
|
│ ├── optimizer.yaml
|
|
|
|
|
|
│ ├── predict.yaml
|
|
|
|
|
|
│ ├── readme
|
|
|
|
|
|
│ └── train.yaml
|
|
|
|
|
|
├── dataset
|
|
|
|
|
|
│ ├── train
|
|
|
|
|
|
│ │ ├── images
|
|
|
|
|
|
│ │ └── labels
|
|
|
|
|
|
│ └── val
|
|
|
|
|
|
│ ├── images
|
|
|
|
|
|
│ └── labels
|
|
|
|
|
|
├── data_utils.py
|
|
|
|
|
|
├── learning_rate_scheduler.py
|
|
|
|
|
|
├── losses.py
|
|
|
|
|
|
├── main.py
|
|
|
|
|
|
├── model.py
|
|
|
|
|
|
├── model_utils.py
|
|
|
|
|
|
├── predict.py
|
|
|
|
|
|
├── pretrained
|
|
|
|
|
|
├── README.md
|
|
|
|
|
|
├── test.ipynb
|
|
|
|
|
|
├── train.py
|
|
|
|
|
|
└── utils.py
|
|
|
|
|
|
|
|
|
|
|
|
10 directories, 19 files
|
|
|
|
|
|
|
|
|
|
|
|
```
|
