office内置gpt
时间: 2023-09-12 08:12:37 浏览: 48
以下是实现代码:
```python
import os
import pandas as pd
from PIL import Image
# 设置数据集路径
data_dir = 'path/to/cat-vs-dog'
# 定义获取图片信息的函数
def get_image_info(file_path):
with Image.open(file_path) as img:
file_name = os.path.basename(file_path)
file_ext = os.path.splitext(file_path)[1]
file_size = os.path.getsize(file_path)
channels = len(img.getbands())
img_size = img.size
return file_name, file_path, file_ext, file_size, channels, img_size
# 遍历数据集文件夹,获取所有图片信息
data = []
for folder in os.listdir(data_dir):
folder_path = os.path.join(data_dir, folder)
if os.path.isdir(folder_path):
for file in os.listdir(folder_path):
file_path = os.path.join(folder_path, file)
if os.path.isfile(file_path):
data.append(get_image_info(file_path))
# 将数据保存到pandas dataframe中
df = pd.DataFrame(data, columns=['file_name', 'file_path', 'file_ext', 'file_size', 'channels', 'img_size'])
# 将数据保存为csv文件
df.to_csv('cat_vs_dog_image_info.csv', index=False)
```
执行后,将在当前目录下生成一个名为"cat_vs_dog_image_info.csv"的文件,其中包含所有图片的信息。