读取所有图片

  1. """
  2. 文件结构
  3. train
  4. ├── 0
  5. |   ├── 1.png
  6. |   ├── 2.png
  7. |   └── 3.png
  8. └── 1
  9.     ├── 4.png
  10.     ├── 5.png
  11.     └── 6.png
  12.     ...
  13. """
  15. train_jpg = glob.glob('./train/*/*')
  17. """
  18. ['./train//0/1.png'
  19. './train//0/2.png'
  20. './train//0/3.png' ...
  21. './train//1/4.png'
  22. './train//1/5.png'
  23. './train//1/6.png']
  24. """

图片尺寸分析

  1. from pathlib import Path
  2. from PIL import Image
  3. import numpy as np
  4. import matplotlib.pyplot as plt
  6. p = Path("./data/GHIM-20/")
  7. img_list = p.glob("[0-9]*/*.jpg")
  9. w_h = []
  10. channel = []
  11. for img in img_list:
  12.     arr = np.array(Image.open(str(img)))
  13.     shape = arr.shape
  14.     w_h.append(list(shape[:-1]))
  15.     channel.append(shape[-1])
  17. print("channel", set(channel))
  18. w_h = np.array(w_h)
  19. plt.scatter(w_h[:,0], w_h[:,1])
  20. plt.show()