diff --git a/README.md b/README.md index 0d3f262..a68e153 100644 --- a/README.md +++ b/README.md @@ -1,5 +1,5 @@ # 使用Pytorch简单实现SegNext,学习用 -pretrained文件夹内由官方训练的tiny版本权重文件 +pretrained文件夹内放置权重文件 可以使用官方版本的tiny权重文件进行训练或者预测(其他版本未尝试) 更改运行参数请使用config文件夹内的yaml文件配置 训练移步train.py,预测移步predict.py, 有不会传的参数请移步main.py