数据集处理实验2——统计图片的长边和短边

有时候想想觉得发一篇文章挺难的，但是自己发文章自己做实验是真香。就这样一点一点来吧。

这个就是统计图片的尺寸大小 然后对数据集进行处理
感谢大佬

程序代码

import os
import pandas as pd
from PIL import Image
from collections import Counter


source = 'G:\pythonproject\image'
imgfile = os.listdir(source)
shot_cut = []
long_cut = []

print('filenum:', len([lists for lists in os.listdir(source)]))

for img in imgfile:
    img_path = os.path.join(source, img)
    img = Image.open(img_path)
    imgSize = img.size  # 图片的长和宽
    #print(imgSize)
    maxSize = max(imgSize)  # 图片的长边
    long_cut.append(maxSize)
    minSize = min(imgSize)  # 图片的短边
    shot_cut.append(minSize)

#shot_result = Counter(shot_cut)
#long_result = Counter(long_cut)
shot_result = pd.value_counts(shot_cut, normalize=True)
long_result = pd.value_counts(long_cut, normalize=True)

print(shot_result)
print(long_result)

shot_result.to_csv('shot.csv', header=0)
long_result.to_csv('long.csv', header=0)

运行结果


CSV文件

长边

短边