ModelZooStd.md 2.01 KB
Newer Older
liuhy's avatar
liuhy committed
1
# 仓库目录结构
chenzk's avatar
v1.0  
chenzk committed
2
3
* 除预训练模型外其他文件总大小尽量不要超过50M
```
liuhy's avatar
liuhy committed
4
    Project
chenzk's avatar
v1.0  
chenzk committed
5
6
7
8
9
10
11
12
13
    ├── dataset
    │   ├── label_1
    │             ├── xxx.png
    │             ├── xxx.png
    │             └── ...
    │   └── label_2
    │             ├── xxx.png
    │             ├── xxx.png
    │             └── ...  
liuhy's avatar
liuhy committed
14
15
16
17
    ├── model
    │   ├── xxx.pth #预训练模型 
    │   ├── xxx.onnx #对应的onnx模型
    │   └── xxx.mxr #对应的migraphx离线推理模型
chenzk's avatar
v1.0  
chenzk committed
18
19
20
21
    ├── doc
    │   ├── icon.png
    │   ├── xxx.png
    │   └── xxx.png
liuhy's avatar
liuhy committed
22
23
24
25
26
27
28
29
30
31
    ├── README.md
    ├── requirement.txt
    ├── model.properties
    ├── code_file1.py
    ├── code_file2.py
    ├── code_file3.py
    ├── dirs
    │   ├── code_file4.py
    │   ├── code_file5.py
    └── └── code_file6.py
chenzk's avatar
v1.0  
chenzk committed
32
33
34
35
36
37
38
39
40
41
```
* icon.png:模型的图标文件,可到[iconfont](https://www.iconfont.cn/?spm=a313x.7781069.1998910419.d4d0a486a)查找。
![img](./doc/icon.png)
* README.md:参照下图,`十二大标题`为必选项,二级标题以下的标题或内容根据自己的项目灵活增删。
![img](./doc/readme.png)
* requirement.txt:模型依赖统一写到此文件,与深度学习相关的库请注释,以免安装为nv库。
* model.properties:`五大属性`固定模板如下:
```
    # 模型唯一标识
    modelCode=Project ID
42
    # 模型名称
chenzk's avatar
v1.0  
chenzk committed
43
    modelName=模型名称(同项目名称:模型名_深度学习框架)
44
    # 模型描述
liuhy's avatar
liuhy committed
45
    modelDescription=简要描述此模型(尽量50字以内)
46
    # 应用场景
chenzk's avatar
v1.0  
chenzk committed
47
    appScenario=推理,训练,OCR,政府,交通,零售,金融,医疗(首先描述推理、训练信息,然后描述算法类别信息,最后描述应用行业信息,多个标签用英文逗号隔开。)
48
    # 框架类型
chenzk's avatar
v1.0  
chenzk committed
49
50
    frameType=paddle(说明使用的算法框架, 多个标签用英文逗号隔开。)
```