2023-04-07 22:40:24 +08:00

333 B
Raw Blame History

使用Pytorch简单实现SegNext学习用

pretrained文件夹内放置权重文件 可以使用官方版本的tiny权重文件进行训练或者预测其他版本未尝试 更改运行参数请使用config文件夹内的yaml文件配置 训练移步train.py预测移步predict.py, 有不会传的参数请移步main.py