笔记：数据处理的代码小技巧（1）

最近觉得自己的基本功好差，要好好学一下如何优雅的写代码。

在深度学习处理中，图片大小是一个很重要的问题，关系着转为tensor的shape。如何方便的查看一个文件中几千张图片的大小。

放在以前我一直只会用for循环，用cv2读取图片，然后直接print(img.shape)，凭借肉眼观察，图片的大小。

最近读到了一个很简单的代码块，可以有效显示，所有图片的大小，以及每种大小的有几张。

shape_dic = {}
for image_name in os.listdir(文件地址）[:300]#查看前300张
    image = cv2.imread(image_name)
    shape_dic[image.shape] = shape_dic.get(image.shape,0) + 1

print(shape_dic)

提示，dictionary.get（a,b)为查找字典中key为a的值，如果不存在，则返回b值。