这篇文章是用来教如何使用opencv来修改图片的大小,在机械学习上很常用到。

如果还想做其他处理可以在下面自己修改一部份的code就可以了。    

# -*- coding: utf-8 -*-
import os
import cv2


def main():
    data_dir_path = u"directory的位置"
    file_list = os.listdir(r'./directory的位置')
    count = 0
    for file_name in file_list:
        root, ext = os.path.splitext(file_name)
        if ext == u'.png' or u'.jpeg' or u'.jpg':
            abs_name = data_dir_path + '/' + file_name
            image = cv2.imread(abs_name)
            #在下面写要做的处理
            img = cv2.resize(image, (224, 224)) #把图片的大小改成224*224
            cv2.imwrite(abs_name, img)          
if __name__ == '__main__':
    main()
查看原文 >>
相关文章