记录一下常使用的代码段.这个是用于从文件夹中随机读取一张图片或者一个文件.
使用时只需要更改get_picture_dir("/home/wsb/桌面/test") 中的地址.读取文件时将图片读取和显示的代码注释掉.
import cv2
import os
from random import randint
def get_picture_dir(file_dir):
filelist = os.listdir(file_dir) #获取文件路径
total_num=len(filelist)
idx = randint(0,total_num)
image_path=filelist[idx]
picture_path = os.path.join(os.path.abspath(file_dir), image_path)
img=cv2.imread(picture_path)
cv2.imshow("img",img)
cv2.waitKey(0)
get_picture_dir("/home/wsb/桌面/test")