mindspore-Ecosystem
/
mindspore
21
0
Fork 27
Code Issues Pull Requests 6 Projects Releases Wiki Activity

update model_zoo/research/cv/vgg19/preprocess.py. 

Add `create_label` description
This commit is contained in:
kingxian 2021-08-17 10:38:02 +00:00 committed by Gitee
parent de43ed5420
commit 92ae657d23
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
model_zoo/research/cv/vgg19/preprocess.py
View File

@ -22,6 +22,7 @@ from model_utils.moxing_adapter import config
def create_label(result_path, dir_path): def create_label(result_path, dir_path):
"""create label for dataset"""
print("[WARNING] Create imagenet label. Currently only use for Imagenet2012!") print("[WARNING] Create imagenet label. Currently only use for Imagenet2012!")
dirs = os.listdir(dir_path) dirs = os.listdir(dir_path)
file_list = [] file_list = []