Upload New File

人工智能系统实战第三期/实战代码/深度学习项目实战/扩散模型作业/DDPM/txt_annotation.py

+import os
+import os
+
+#---------------------------------#
+#   指向数据集所在的文件夹
+#   默认指向根目录下的datasets
+#---------------------------------#
+datasets_path   = "datasets/"
+
+if __name__ == "__main__":
+    photos_names    = os.listdir(datasets_path)
+    photos_names    = sorted(photos_names)
+
+    list_file       = open('train_lines.txt', 'w')
+    for photo_name in photos_names:
+        if(photo_name.lower().endswith(('.bmp', '.dib', '.png', '.jpg', '.jpeg', '.pbm', '.pgm', '.ppm', '.tif', '.tiff'))):
+            list_file.write(os.path.join(os.path.abspath(datasets_path), photo_name))
+            list_file.write('\n')
+    list_file.close()
+