42 lines
1.1 KiB
Markdown
42 lines
1.1 KiB
Markdown
# 使用Pytorch简单实现SegNext，学习用
|
||
pretrained文件夹内放置权重文件
|
||
可以使用官方版本的tiny权重文件进行训练或者预测（其他版本未尝试）
|
||
更改运行参数请使用config文件夹内的yaml文件配置
|
||
训练移步train.py，预测移步predict.py, 有不会传的参数请移步main.py
|
||
|
||
文件应当按照如下形式组织
|
||
|
||
```
|
||
.
|
||
├── 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
|
||
|
||
``` |