cv2.imwrite(os.path.join(img_root, 'photo4', filename), erosion) cv2.error: OpenCV(4.7.0) D:\a\opencv-python\opencv-python\opencv\modules\imgcodecs\src\loadsave.cpp:692: error: (-2:Unspecified error) could not find a writer for the specified extension in function 'cv::imwrite_'
时间: 2023-10-16 12:58:40 浏览: 160
这个错误通常是因为你使用的图像格式不支持OpenCV的imwrite函数。请检查你指定的文件扩展名是否正确,例如,如果你指定的文件扩展名为".jpg",则确保你的图像可以保存为JPEG格式。另外,请确保你的OpenCV版本支持你想要保存的图像格式。你可以查看OpenCV文档中的imwrite函数,以了解支持的图像格式和相应的文件扩展名。
相关问题
from pdb import set_trace as st import os import numpy as np import cv2 import argparse parser = argparse.ArgumentParser('create image pairs') parser.add_argument('--fold_A', dest='fold_A', help='input directory for image A', type=str, default='../dataset/50kshoes_edges') parser.add_argument('--fold_B', dest='fold_B', help='input directory for image B', type=str, default='../dataset/50kshoes_jpg') parser.add_argument('--fold_AB', dest='fold_AB', help='output directory', type=str, default='../dataset/test_AB') parser.add_argument('--num_imgs', dest='num_imgs', help='number of images',type=int, default=1000000) parser.add_argument('--use_AB', dest='use_AB', help='if true: (0001_A, 0001_B) to (0001_AB)',action='store_true') args = parser.parse_args() for arg in vars(args): print('[%s] = ' % arg, getattr(args, arg)) splits = os.listdir(args.fold_A) for sp in splits: img_fold_A = os.path.join(args.fold_A, sp) img_fold_B = os.path.join(args.fold_B, sp) img_list = os.listdir(img_fold_A) if args.use_AB: img_list = [img_path for img_path in img_list if '_A.' in img_path] num_imgs = min(args.num_imgs, len(img_list)) print('split = %s, use %d/%d images' % (sp, num_imgs, len(img_list))) img_fold_AB = os.path.join(args.fold_AB, sp) if not os.path.isdir(img_fold_AB): os.makedirs(img_fold_AB) print('split = %s, number of images = %d' % (sp, num_imgs)) for n in range(num_imgs): name_A = img_list[n] path_A = os.path.join(img_fold_A, name_A) if args.use_AB: name_B = name_A.replace('_A.', '_B.') else: name_B = name_A path_B = os.path.join(img_fold_B, name_B) if os.path.isfile(path_A) and os.path.isfile(path_B): name_AB = name_A if args.use_AB: name_AB = name_AB.replace('_A.', '.') # remove _A path_AB = os.path.join(img_fold_AB, name_AB) im_A = cv2.imread(path_A, cv2.IMREAD_COLOR) im_B = cv2.imread(path_B, cv2.IMREAD_COLOR) im_AB = np.concatenate([im_A, im_B], 1) cv2.imwrite(path_AB, im_AB),解释上述代码,并告诉我怎么设置文件夹格式
这段代码用于创建图像对,将两个文件夹中的图像拼接在一起,并将结果保存在一个新的文件夹中。下面是代码的简要解释:
1. 引入必要的库:pdb、os、numpy、cv2和argparse。
2. 使用argparse来解析命令行参数,包括输入目录和输出目录等。
3. 遍历输入目录中的所有文件夹,并获取每个文件夹中的图像列表。
4. 根据命令行参数,选择要使用的图像对(如果--use_AB参数为true,则仅使用"_A"结尾的图像作为输入A,使用相应的"_B"图像作为输入B)。
5. 将输入A和输入B图像拼接在一起,并将结果保存在输出目录中。
6. 最后,打印出图像对的数量和输出目录等信息。
文件夹格式应该是这样的:
- dataset
- 50kshoes_edges
- split1
- 0001_A.png
- 0002_A.png
- ...
- split2
- 0001_A.png
- 0002_A.png
- ...
- ...
- 50kshoes_jpg
- split1
- 0001_B.jpg
- 0002_B.jpg
- ...
- split2
- 0001_B.jpg
- 0002_B.jpg
- ...
- ...
- test_AB
- split1
- 0001.png
- 0002.png
- ...
- split2
- 0001.png
- 0002.png
- ...
- ...
翻译代码def resize(): img_path = "Normalresult" save_path = "result" img_names = os.listdir(img_path) # 以列表的形式获取文件夹中的所有文件的名字和格式(例如:0.jpg) for img_name in img_names: img = os.path.join(img_path, img_name) # 将文件的绝对路径与每个文件名字进行拼接,以获取该文件 img = cv2.imread(img) # 读取该文件(图片) # 一是通过设置图像缩放比例,即缩放因子,来对图像进行放大或缩小 res1 = cv2.resize(img, None, fx=2, fy=2, interpolation=cv2.INTER_LINEAR) height, width = img.shape[:2] # 二是直接设置图像的大小,不需要缩放因子 res2 = cv2.resize(img, (int(0.8 * width), int(0.8 * height)), interpolation=cv2.INTER_AREA) dir_name1 = ''.join(random.sample(string.ascii_letters + string.digits, 8)) dir_name2 = ''.join(random.sample(string.ascii_letters + string.digits, 8)) cv2.imwrite(os.path.join(save_path, dir_name1 + '.jpg'), res1) cv2.imwrite(os.path.join(save_path, dir_name2 + '.jpg'), res2) cv2.imwrite(os.path.join(save_path, img_name), img)
这段代码的作用是对指定路径下的所有图片进行缩放,并将缩放后的图片保存到指定的保存路径中。具体来说,代码首先使用 os 库中的 listdir 函数获取指定路径下所有图片的文件名,并使用循环遍历每个图片文件名。在循环中,代码使用 os 库中的 join 函数将指定路径和当前图片文件名拼接成完整的路径,并使用 cv2 库中的 imread 函数读取当前图片。
接着,代码使用 cv2 库中的 resize 函数对当前图片进行两种缩放方式的处理。第一种方式是通过设置缩放因子 fx 和 fy 来对图像进行放大或缩小,得到缩放后的图像 res1。第二种方式是直接设置缩放后的图像的大小,得到缩放后的图像 res2。
然后,代码使用 random 和 string 库中的函数生成随机的文件名,并使用 cv2 库中的 imwrite 函数将缩放后的图像 res1 和 res2 以及原始图像 img 分别保存到指定的保存路径中。最后,代码继续循环处理下一个图片。
阅读全文