未验证 提交 feff2cde 编写于 作者: L Liufang Sang 提交者: GitHub

fix bug from QA (#89)

上级 f8a4b1cb
...@@ -156,8 +156,8 @@ def _reader_creator(file_list, ...@@ -156,8 +156,8 @@ def _reader_creator(file_list,
for line in lines: for line in lines:
if mode == 'train' or mode == 'val': if mode == 'train' or mode == 'val':
img_path, label = line.split() img_path, label = line.split()
img_path = os.path.join(data_dir + "/" + mode, img_path = os.path.join(
img_path) os.path.join(data_dir, mode), img_path)
yield img_path, int(label) yield img_path, int(label)
elif mode == 'test': elif mode == 'test':
img_path = os.path.join(data_dir, line) img_path = os.path.join(data_dir, line)
......
...@@ -10,7 +10,7 @@ ...@@ -10,7 +10,7 @@
### 准备数据 ### 准备数据
``demo``文件夹下创建``data``文件夹,将``ImageNet``数据集解压在``data``文件夹下,解压后``data``文件夹下应包含以下文件: ``demo``文件夹下创建``data``文件夹,将``ImageNet``数据集解压在``data``文件夹下,解压后``data/ILSVRC2012``文件夹下应包含以下文件:
- ``'train'``文件夹,训练图片 - ``'train'``文件夹,训练图片
- ``'train_list.txt'``文件 - ``'train_list.txt'``文件
- ``'val'``文件夹,验证图片 - ``'val'``文件夹,验证图片
......
...@@ -10,7 +10,7 @@ ...@@ -10,7 +10,7 @@
### 准备数据 ### 准备数据
``demo``文件夹下创建``data``文件夹,将``ImageNet``数据集解压在``data``文件夹下,解压后``data``文件夹下应包含以下文件: ``demo``文件夹下创建``data``文件夹,将``ImageNet``数据集解压在``data``文件夹下,解压后``data/ILSVRC2012``文件夹下应包含以下文件:
- ``'train'``文件夹,训练图片 - ``'train'``文件夹,训练图片
- ``'train_list.txt'``文件 - ``'train_list.txt'``文件
- ``'val'``文件夹,验证图片 - ``'val'``文件夹,验证图片
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册