拡張子のついていない画像群に一括で拡張子をつけたい

Question

機械学習用にダウンロードした画像のデータセットに拡張子がついていなかったので拡張子をつけたいです。
具体的には、datasetというフォルダの中に学習用が10個、検証用とテスト用が1個ずつという感じです。
何度も実行するのは避けたいので、何とか学習用、検証用、テスト用の画像に一括で拡張子つけたいと考えていますが上手くいきませんので教えていただきたいです。

import os
from pathlib import Path
from PIL import Image
import cv2
import numpy as np

def get_paths(input_dir):
    paths = sorted([x for x in input_dir.iterdir()])


    return paths

# ディレクトリ指定
input_dir = Path(r"./dataset/")


#---------- 指定フォルダ内のファイル拡張子を変更 ----------#
for f in get_paths(input_dir):
   f.rename(input_dir + '.jpg')

Answer 1

datasetというフォルダの中にサブフォルダがあるらしいのでそれに対応しました。

import os

def add_file_ext_in_folder(input_dir, ext):
    for name in os.listdir(input_dir):
        obj = os.path.join(input_dir, name)
        if os.path.isdir(obj):
            add_file_ext_in_folder(obj, ext)
        elif os.path.isfile(obj):
            os.rename(obj, obj + ext)

input_dir = r"./dataset/"
add_file_ext_in_folder(input_dir, '.jpg')

以前のコードは以下です。

import os

def add_file_ext_in_folder(input_dir, ext):
    for name in os.listdir(input_dir):
        fname = os.path.join(input_dir, name)
        os.rename(fname, fname + ext)

input_dir = r"./dataset/"
add_file_ext_in_folder(input_dir, '.jpg')

Comments

[xeno]

申し訳ありません。
datasetというディレクトリ内にある学習用、検証用、テスト用のディレクトリに保存されている画像に対して拡張子を付与したいです。
教えて頂いたものだと、学習用、検証用、テスト用のディレクトリに対して拡張子が付与されてしまいます。

[lehshell]

datasetというディレクトリ内は画像ファイルだけと読み取ってしまいました。
サブフォルダに対応したコードに更新しました。

[xeno]

追加の要望にも応えて頂きありがとうございました