Update whl.md

pull/692/head
littletomatodonkey 2021-04-20 14:00:28 +08:00 committed by GitHub
parent cf880114d1
commit fec5b5147e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 4 deletions
docs/zh_CN

View File

@ -49,9 +49,6 @@ paddleclas --model_name=ResNet50 --top_k=5 --image_file='docs/images/whl/demo.jp
Predict complete!
```
**注意**: 如果使用`Transformer`系列模型,如`DeiT_***_384`, `ViT_***_384`等,请注意模型的输入数据尺寸,需要设置参数`resize_short=384`, `reize=384`
### 2. 参数解释
以下参数可在命令行交互使用时通过参数指定或在Python代码中实例化PaddleClas对象时作为构造函数的参数使用。
* model_name(str): 模型名称没有指定自定义的model_file和params_file时可以指定该参数使用PaddleClas提供的基于ImageNet1k的inference model默认值为ResNet50。
@ -75,7 +72,11 @@ paddleclas --model_name=ResNet50 --top_k=5 --image_file='docs/images/whl/demo.jp
* pre_label_image(bool): 是否需要进行预标注。
* pre_label_out_idr(str): 进行预标注后输出结果的文件路径默认为None。
**注意**: 如果使用`Transformer`系列模型,如`DeiT_***_384`, `ViT_***_384`等,请注意模型的输入数据尺寸,需要设置参数`resize_short=384`, `reize=384`
**注意**: 如果使用`Transformer`系列模型,如`DeiT_***_384`, `ViT_***_384`等,请注意模型的输入数据尺寸,需要设置参数`resize_short=384`, `reize=384`,如下所示。
```
clas = PaddleClas(model_name='ViT_base_patch16_384', top_k=5, resize_short=384, resize=384)
```
### 3. 代码使用方法