Python随缘选择图片

 

import cv2
import os
import random
import matplotlib.pyplot as plt

#读取当前目录dir下的所有图片路径
def all_file_path(dir):
    all_path = []
    pth = os.path.abspath(os.path.dirname(__file__) + '/' + dir)
    print(pth)
    for maindir, subdir, file_list in os.walk(pth):
        pass
    for filename in file_list:
        all_path.append(os.path.join(pth,filename))
    return all_path


def show_all_image(path_lst):
    index = 0
    for path in path_lst:
        img = cv2.imread(path)
        title = 'title' + str(index + 1)
        plt.subplot(3, 3, index + 1)
        plt.imshow(img)
        plt.title(title, fontsize=8)
        plt.xticks([])
        plt.yticks([])
        index += 1
    plt.show()



path_lst = all_file_path('Image')  #代码目录下建一个Image文件夹存放图片
#show_all_image(path_lst)
path = random.choice(path_lst)      #随机选择目录 ，不要太简单
print(path)
img = cv2.imread(path)
cv2.imshow('The best image in all', img)     #show 图片 set Title
cv2.waitKey()