更新 README.md,添加文件组织结构示例
This commit is contained in:
parent
4799da54ce
commit
294425fce4
37
README.md
37
README.md
@ -3,3 +3,40 @@ 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
|
||||
|
||||
```
|
Loading…
x
Reference in New Issue
Block a user