Python3 作成したnpyファイルの保存先をカレントディレクトリ直下にしたい

Question

前提・実現したいこと

画像をnumpy配列に変換し、その後npyファイルに保存する処理を１００回繰り返すfor文を作成しました。
コードは一応動いたのですが、作成されたnpyファイルが、カレントディレクトリ直下ではなく、その下のtrainフォルダ内に作成されてしまいます。
できればカレントディレクトリ直下に保存されてほしいのですが、コードをどのように修正すれば良いかご教示頂けないでしょうか。

該当のソースコード

img_size = (1000,500)
dir_name ='./train' 
file_type = 'jpeg'

img_list = glob.glob('./' + dir_name + '/*.' + file_type)

for i in range(100): 
    train_img_array_list = []
    rn.sample(img_list, 50) #iが1増加するたび、ランダムに選択した５０枚の画像を読み込み
    for img in img_list:
      train_img = load_img(img,grayscale=True,target_size=(img_size))  
      train_img_array = img_to_array(train_img) /255
      train_img_array_list.append(train_img_array)
      #print(train_img_array_list)

    train_img_array_list = np.array(train_img_array_list)
    file_name = os.path.join(dir_name, 'image'+'{0:04d}'.format(i)+'.npy')
    np.save(file_name, train_img_array_list)

試したこと

np.saveの一文の前にos.chdir文を追加してディレクトリを指定してみたのですが、変わりませんでした。

Answer 1

出力ファイルの名前を以下の行で作っていますが、この先頭のdir_nameが./trainディレクトリになっているのが原因ですね。 dir_nameに代わりに"."とすればできるでしょう。

file_name = os.path.join(dir_name, 'image'+'{0:04d}'.format(i)+'.npy')

Comments

[SuzuAya]

お忙しいところ，早速ご回答頂きありがとうございました！
無事カレントディレクトリ直下に保存する事ができました。大変助かりました。