def generate_train(path, result, filters, train): files = os.listdir(path) for f in files: train_img = cv.imread(path + '/' + f, cv.IMREAD_GRAYSCALE) res = Gabor().getGabor(train_img, filters) res = np.array(res).reshape(-1) res = np.append(res, result) train.append(res)
时间: 2024-04-25 15:27:58 浏览: 93
这是一个函数,用于生成训练数据。函数需要四个参数:path表示训练集图片所在的路径,result表示训练集图片的标签,filters表示Gabor滤波器的参数,train表示用于存储生成的训练数据。
该函数首先列出指定路径下的所有图片,接着读取每个图片,并使用Gabor算法对其进行滤波,得到一个包含多个特征的向量。然后将这个向量与该图片的标签合并为一个样本,并将其添加到train中。最后,函数返回train列表,其中包含了所有生成的训练数据。
相关问题
import os import cv2 # 定义原始和更改大小后的文件夹路径 original_folder_path = 'E:\shujuji\Intel Image Classification\seg_train\seg_train/buildings' resized_folder_path = 'E:\shujuji\Intel Image Classification\seg_train\change/buildings' # 如果更改大小后的文件夹不存在,则创建它 if not os.path.exists(resized_folder_path): os.makedirs(resized_folder_path) # 循环遍历原始文件夹中的所有图像文件 for filename in os.listdir(original_folder_path): if filename.endswith('.jpg') or filename.endswith('.png') or filename.endswith('.jpeg'): # 读取原始图像 img = cv2.imread(os.path.join(original_folder_path, filename)) # 将图像大小更改为(224,224,3) img_resized = cv2.resize(img, (224, 224)) # 保存更改大小后的图像到新文件夹中,使用原始文件名 cv2.imwrite(os.path.join(resized_folder_path, filename), img_resized),对这段代码进行更改,使之满足要求
以下是修改后的代码,满足获取文件夹下所有文件夹的所有图片,更改图片的尺寸后生成文件夹,按照原始文件夹进行排布,文件名使用原始文件名的要求:
```python
import os
import cv2
# 定义函数,用于调整图片尺寸
def resize_image(image_path, output_path, new_size):
img = cv2.imread(image_path)
img_resized = cv2.resize(img, new_size)
cv2.imwrite(output_path, img_resized)
# 定义函数,用于获取指定文件夹下所有图片
def get_all_images(folder_path):
images = []
for root, dirs, files in os.walk(folder_path):
for file in files:
if file.endswith('.jpg') or file.endswith('.png') or file.endswith('.jpeg'):
images.append(os.path.join(root, file))
return images
# 定义函数,用于生成新文件夹并存储调整后的图片
def generate_resized_images(source_folder, target_folder, new_size):
# 获取所有图片
images = get_all_images(source_folder)
# 按照原始文件夹进行排布
for image_path in images:
relative_path = os.path.relpath(image_path, source_folder)
output_folder = os.path.join(target_folder, os.path.dirname(relative_path))
os.makedirs(output_folder, exist_ok=True)
output_path = os.path.join(output_folder, os.path.basename(image_path))
# 调整图片尺寸并存储
resize_image(image_path, output_path, new_size)
# 定义原始和更改大小后的文件夹路径
original_folder_path = r'E:\shujuji\Intel Image Classification\seg_train\seg_train\buildings'
resized_folder_path = r'E:\shujuji\Intel Image Classification\seg_train\change\buildings'
# 如果更改大小后的文件夹不存在,则创建它
if not os.path.exists(resized_folder_path):
os.makedirs(resized_folder_path)
# 调用函数,生成新文件夹并存储调整后的图片
new_size = (224, 224) # 调整后的图片尺寸为 224x224
generate_resized_images(original_folder_path, resized_folder_path, new_size)
```
在使用时,需要将代码中的 `original_folder_path` 替换为要处理的原始文件夹的路径,`resized_folder_path` 替换为生成的新文件夹的路径,`new_size` 替换为要调整的图片尺寸。代码会自动获取原始文件夹下所有图片,并按照原始文件夹进行排布,生成新文件夹并存储调整后的图片。新文件夹的文件名使用原始文件名。
阅读全文