瀏覽代碼

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

liyan 1 年之前
父節點
當前提交
4ea0a02b0c
共有 1 個文件被更改,包括 1 次插入0 次删除
  1. 1 0
      tool/generate_txt.py

+ 1 - 0
tool/generate_txt.py

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