소스 검색

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

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->分类名称->图片
 '''