Pārlūkot izejas kodu

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

liyan 1 gadu atpakaļ
vecāks
revīzija
4ea0a02b0c
1 mainītis faili ar 1 papildinājumiem un 0 dzēšanām
  1. 1 0
      tool/generate_txt.py

+ 1 - 0
tool/generate_txt.py

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