Browse Source

生成标签文本添加数据集文件组成注释

liyan 1 year ago
parent
commit
4ea0a02b0c
1 changed files with 1 additions and 0 deletions
  1. 1 0
      tool/generate_txt.py

+ 1 - 0
tool/generate_txt.py

@@ -3,6 +3,7 @@ from torchvision import datasets
 
 '''
     为数据集生成对应的txt文件
+    数据集排列格式:数据集名称->train/test->分类名称->图片
 '''