這篇文章是用來教如何使用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()
查看原文 >>
相關文章