p19647583
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

add requirements for unet

This commit is contained in:
zhaoting 2021-04-07 10:43:55 +08:00
parent c543e121ed
commit 17f5f4be92
4 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
model_zoo/official/cv/unet/README.md
View File

@ -144,6 +144,7 @@ Then you can run everything just like on ascend.
├── mindspore_hub_conf.py // hub config file
├── postprocess.py // unet 310 infer postprocess.
├── preprocess.py // unet 310 infer preprocess dataset
├── requirements.txt // Requirements of third party package.
```
### [Script Parameters](#contents)

1
model_zoo/official/cv/unet/README_CN.md
View File

@ -148,6 +148,7 @@ bash scripts/docker_start.sh unet:20.1.0 [DATA_DIR] [MODEL_DIR]
├── mindspore_hub_conf.py // hub 配置脚本
├── postprocess.py // 310 推理后处理脚本
├── preprocess.py // 310 推理前处理脚本
├── requirements.txt // 需要的三方库.
```
### 脚本参数

2
model_zoo/official/cv/unet/requirements.txt Normal file
View File

@ -0,0 +1,2 @@
opencv-python >= 4.1.2.30
pillow >= 6.2.0

2
model_zoo/official/nlp/tinybert/export.py
View File

@ -48,7 +48,7 @@ task_params = {"SST-2": {"num_labels": 2, "seq_length": 64},
"QNLI": {"num_labels": 2, "seq_length": 128},
"MNLI": {"num_labels": 3, "seq_length": 128},
"TNEWS": {"num_labels": 15, "seq_length": 128},
"CLUENER": {"num_labels": 10, "seq_length": 128}}
"CLUENER": {"num_labels": 43, "seq_length": 128}}
class Task:
"""