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

Add `create_label` description
2021-08-17 10:38:02 +00:00 · 2021-08-17 10:38:02 +00:00 · 92ae657d23
parent de43ed5420
commit 92ae657d23
1 changed files with 1 additions and 0 deletions
--- a/model_zoo/research/cv/vgg19/preprocess.py
+++ b/model_zoo/research/cv/vgg19/preprocess.py
@ -22,6 +22,7 @@ from model_utils.moxing_adapter import config


 def create_label(result_path, dir_path):
+    """create label for dataset"""
    print("[WARNING] Create imagenet label. Currently only use for Imagenet2012!")
    dirs = os.listdir(dir_path)
    file_list = []