python读取文件夹中所有的图片和视频
时间: 2024-11-09 09:24:09 浏览: 8
在Python中,我们可以使用`os`和`glob`模块来遍历文件夹并读取其中的所有图片和视频。以下是基本步骤:
1. 导入需要的库:
```python
import os
from PIL import Image
import cv2
```
这里我们引入了`os`来处理文件路径,`PIL`用于处理图片,`cv2`用于处理视频。
2. 定义一个函数来检查文件是否为图片或视频:
```python
def is_image_or_video(file_path):
if file_path.endswith(('.jpg', '.jpeg', '.png', '.gif')):
return True
elif file_path.endswith(('.mp4', '.avi', '.mkv')):
try:
vidcap = cv2.VideoCapture(file_path)
vidcap.release()
return True
except cv2.error:
pass
return False
```
这个函数会检查文件的扩展名,如果是常见的图片格式(如.jpg、.png等),或者视频格式(如.mp4、.avi等),则返回True。
3. 遍历文件夹:
```python
folder_path = 'your_folder_directory' # 替换为你想要扫描的文件夹路径
for filename in os.listdir(folder_path):
file_path = os.path.join(folder_path, filename)
if is_image_or_video(file_path):
print(f"Found file: {file_path}")
if is_image_file := is_image_or_video(file_path): # 对图片进行操作...
img = Image.open(file_path) # 使用PIL打开图片
# ...进行其他图片处理
elif is_video_file := is_image_or_video(file_path): # 对视频进行操作...
cap = cv2.VideoCapture(file_path) # 使用cv2打开视频
# ...进行其他视频处理
```
在这个例子中,你需要自行添加对图片和视频的具体处理部分。
阅读全文