自己写了个采样的代码!因为我的图片在代码的上一级文件夹,就用了root_directory,各位记得改路径哈!
import os
import cv2
import sys
# print(os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))
sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '..'))) # 返回上一级目录,返回至根目录
def sample(fil_path, sav_path, sampling_num):
file_list = os.listdir(fil_path) # 获取文件列表
print("文件的个数为:", len(file_list))
for item in file_list:
if item.endswith('.jpg'):
path = os.path.join(os.path.abspath(fil_path), item) # 添加路径以及图片的名字
img = cv2.imread(path) # 读取图片 必须加上路径,否则图片加载容易出问题
if sampling_num > 1:
for i in range(sampling_num - 1):
img = cv2.pyrUp(img) # 上采样
# img = cv2.pyrDown(img) # 下采样
elif sampling_num == 1:
img = img
el