機械学習のデータセットのためのファイル名一覧などに。


import os

path = "./datasets/train" # ファイル名一覧を取得したいディレクトリのパス
text_file = open("./datasets/train.txt", "w") # 書き込み先のテキストファイルを作る

files = os.listdir(path) # ファイル名一覧を取得

for f in files:
  basename_without_ext,ext = os.path.splitext(os.path.basename(f)) # 拡張子抜きのファイル名を取得
  text_file.write(basename_without_ext + "\n") # テキストファイルに書き込む

text_file.close() # テキストファイルを保存

image0
image1
image2
image3
・
・
・

という感じで書き込まれます。
拡張子も書き込みたい場合は、


 # basename_without_ext,ext = os.path.splitext(os.path.basename(f)) # 拡張子抜きのファイル名を取得
basename = os.path.basename(f) # 拡張子ありのファイル名を取得

image0.jpg
image1.jpg
image2.jpg
image3.jpg
・
・
・

🐣

お仕事のご相談こちらまで
rockyshikoku@gmail.com

Core MLを使ったアプリを作っています。
機械学習関連の情報を発信しています。

Twitter
Medium