複数読み込んだ画像をリサイズし同じ名前で保存したい

Question

やりたいこと

・フォルダから複数の画像を取得
・リサイズ
・ファイル名の同じ名前で保存したい　←ここです。

例えば

.data/cat/の中に
00001.jpg
00002.jpg
・
・
・
と続く画像を以下のコードで一枚ずつリサイズします。
それらを同じ名前で000001.jpg、000002.jpgと一枚ずつ保存したいです。

環境

python3.7
opencv 4.2
wsl 1(ubuntu18.04)
windows 10 home

書いたコード

import os
import glob
import cv2
# 画像を入力
imgs = glob.glob('./data/cat/*.jpg')


def trans_square(imgs):

    for i,img_list in enumerate(imgs):
        img = cv2.imread(img_list)
        resize_img = cv2.resize(img,(32,32))
        h,w,c = resize_img.shape
        cv2.imwrite('/home/output/sample.png',resize_img)
        print("{},{},{}".format(h,w,c))
        if i==0:
            break

    return img




trans_square(imgs)

このままだと当然ながら一枚しか保存できないのです。。。。

試したこと

こちらに似たようなことをされている方がいましたが、初心者ゆえに似たようなコードから自分のコードに落とし込めないでいます。
完全に力不足です。

おねがいしたいこと

上にあるコードをやりたいことにそうような形にする方法やコードのレビューをしていただければと思います。
何か至らない点があればその都度修正いたしますのでよろしくお願いします。

Comments

[TakaiY]

-「ファイル名の同じ名前で保存したい」とはどういう意味でしょうか。
例をあげて具体的に説明してください。
"元画像が「hoge.jpg」のときは「hoge.png」にしたい"
など
- 以下のようになっているのはなぜですか？
if i==0:
break

[yuki23]

OpenCV でやる必要はありますか？ 正直、向いていない用途です

[oinari03]

このときif == 0で止めているのは1~100枚ほど連番で連なった画像をリサイズしているのですが、どのように一枚ずつ違う名前保存すればわからなかったので１番目で止めました。（0001という名前の処理前画像をリサイズしたら0001という名前で保存、2なら2に、3なら3にという感じです...）

今回はopencvでやることが自分の中の課題でどうしても達成したいものです。
正直、初心者過ぎて話の内容についていけない可能性がありますがお許しください。。。

Answer 1

これでどうですか？

import os
import glob
import cv2

imgs = glob.glob('./data/cat/*.jpg')
for i in imgs:
    img = cv2.imread(i)
    resize_img = cv2.resize(img, (32, 32))
    cv2.imwrite(i, resize_img)

Answer 2

以下にて～/0001.jpgのうち0001のみ抽出できます。
参考：Pythonでパス文字列からファイル名・フォルダ名・拡張子を取得、結合

    fname = os.path.splitext(os.path.basename(img_list))[0]
    cv2.imwrite(f'/home/output/{fname}.png',resize_img)

Comments

[oinari03]

ありがとうございます！！参考リンクも助かります！！