使用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