ファイル内の画像の数をカウントしたいです（解決方法を教えてほしいです）

Question

import os
import re

os.system("ls")

direct='C:/Users/eicha/Desktop/test'#画像が入っているディレクトリの指定

info=os.listdir(direct)#ディレクトリの情報を引き出す

count=1

for file in info:
index=re.match('jpg',info)#jpg形式のファイルがinfo内にあるか調べる
if index:
count+=1 #info内にjpgを発見したらカウントが増える

print(count)

とすると、以下のようなエラーが出てきます。

Traceback (most recent call last):
File "C:/Users/eicha/Desktop/YOLOv4 image save.py", line 14, in <module>
index=re.match('jpg',info)#jpg形式のファイルがinfo内にあるか調べる
File "C:\Program Files\Python39\lib\re.py", line 191, in match
return _compile(pattern, flags).match(string)
TypeError: expected string or bytes-like object

このエラーの解決方法を教えてください。

参考サイト：https://blog-ryotaro-diary.hatenablog.com/entry/2018/12/28/222547

Comments

[meg_]

コードは「コードの挿入」で記入してください。

Answer 1

• このエラーの解決方法を教えてください。

index=re.match('jpg',info)

を

index=re.search('jpg',file)

に変更すればこのエラーは解決します。

あと二つの間違いを修正すれば正しい答えが出るようになりますのでがんばってください。

Answer 2

Python3入門 フォルダ内のファイル一覧を取得する方法

ここらへんを参考にされては。