Python opencv 如何獲取圖片列表及圖片大小

woff

先準備一目錄 byfish 裡面放了圖片


代碼如下：

1. import os
   
2. import cv2
   
3. import numpy as np
   
4. 
   
5. pic_floder = 'byfish'
   
6. # open floder
   
7. i = 0
   
8. with open(pic_floder+'.txt', 'w') as f:
   
9.     # search floder pic name
   
10.     for img in os.listdir(pic_floder):
    
11.         # get pic width and height
    
12.         img_hw = cv2.imread(pic_floder+"/"+img)
    
13.         sp = img_hw.shape
    
14. #         print(sp)
    
15. #         sz1 = sp[0]#height(rows) of image
    
16. #         sz2 = sp[1]#width(colums) of image
    
17. #         sz3 = sp[2]#the pixels value is made up of three primary colors
    
18. #         print('width: %d \nheight: %d \nnumber: %d' %(sz1,sz2,sz3))
    
19. #            pic_floder, img_name,img_num,rectX,rectY,rectW,rectH
    
20.         line = pic_floder+'/'+img+'  1    0    0    '+str(sp[1])+'  '+str(sp[0])+'\n'
    
21.         f.write(line)
    
22.         i += 1
    
23. #計算圖片數量
    
24. print(i)

複製代碼

結果會建立 byfish.txt 檔案
打開看看



網頁設計,網站架設 ,網路行銷,網頁優化,SEO - NetYea 網頁設計