From 92ae657d23e5ede5b9c0db82f81bfeea0397ebb8 Mon Sep 17 00:00:00 2001 From: kingxian Date: Tue, 17 Aug 2021 10:38:02 +0000 Subject: [PATCH] update model_zoo/research/cv/vgg19/preprocess.py. Add `create_label` description --- model_zoo/research/cv/vgg19/preprocess.py | 1 + 1 file changed, 1 insertion(+) diff --git a/model_zoo/research/cv/vgg19/preprocess.py b/model_zoo/research/cv/vgg19/preprocess.py index 819b9f1877e..1e72e41fa93 100755 --- 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 = []