将图像写入不同的文件夹
Writing images into a different folder
我有两个目录:
dir1 = path/to/original/imgs
和
dir2 = path/to/subset/imgs
dir1 包含来自 COCO 数据集的图像,dir2 当前为空。我还有一个列表,其中包含一些图像的名称(从 dir1 中提取):
list1 = ['img1', 'img2', 'img3', 'img4', 'img5']
我需要做的是将list1中的图片名称与dir1中的图片名称进行比较,并将匹配的图片保存到dir2中。以下是我到目前为止所拥有的:
import os
path_to_imgs = "/path/to/dir1"
path_to_subset_imgs = "/path/to/dir2"
file_list = os.listdir(path_to_imgs)
for img_name in list1:
for filename in file_list:
if img_name == filename:
我无法理解如何将相同的图像保存到 dir2。我检查了 and this one。非常感谢任何帮助。
谢谢。
您可以考虑使用 shutil
import shutil
image_list = ['img1', 'img2', 'img3', 'img4', 'img5']
dirs_list = [("/path/to/dir1/", "/path/to/dir2/")]
for img in image_list:
for source_folder, destination_folder in dirs_list:
shutil.copy(source_folder+img, destination_folder+img)
首先检查文件夹 dir1 中是否存在图像,如果存在则将图像从 dir1 复制到 dir2。
for img in list1:
if os.path.exists(dir1+img):
shutil.copy(dir1+img, dir2+img)
else:
print "Image %s doesn't exist"%img
我有两个目录:
dir1 = path/to/original/imgs
和
dir2 = path/to/subset/imgs
dir1 包含来自 COCO 数据集的图像,dir2 当前为空。我还有一个列表,其中包含一些图像的名称(从 dir1 中提取):
list1 = ['img1', 'img2', 'img3', 'img4', 'img5']
我需要做的是将list1中的图片名称与dir1中的图片名称进行比较,并将匹配的图片保存到dir2中。以下是我到目前为止所拥有的:
import os
path_to_imgs = "/path/to/dir1"
path_to_subset_imgs = "/path/to/dir2"
file_list = os.listdir(path_to_imgs)
for img_name in list1:
for filename in file_list:
if img_name == filename:
我无法理解如何将相同的图像保存到 dir2。我检查了
您可以考虑使用 shutil
import shutil
image_list = ['img1', 'img2', 'img3', 'img4', 'img5']
dirs_list = [("/path/to/dir1/", "/path/to/dir2/")]
for img in image_list:
for source_folder, destination_folder in dirs_list:
shutil.copy(source_folder+img, destination_folder+img)
首先检查文件夹 dir1 中是否存在图像,如果存在则将图像从 dir1 复制到 dir2。
for img in list1:
if os.path.exists(dir1+img):
shutil.copy(dir1+img, dir2+img)
else:
print "Image %s doesn't exist"%img