把一个文件夹下的部分图像保存到另一个文件夹下

# -*- coding: utf-8 -*-
import os.path
import os
import cv2
#将要转移的图像生成test.txt文本
with open('H:/data/125penguan1/test.txt', 'r') as lines:
    for line in lines:
#将每行中的多余空格删除
        line = line.strip()
#line_是图像路径
        line_ = 'H:/'+line
#读入图像
        img = cv2.imread(line_)
        name = line.split('/')
#要保存部分图像的路径,name[3]是图像名字
        dir_img = 'H:/'+name[0] + '/' +name[1] + '/imgs/'+name[3]
#根据test.txt文本路径读到的图像,将图像保存到dir_img中
        cv2.imwrite(dir_img, img)
lines.close()

cv2保存图像时必须给图像名字。

你可能感兴趣的:(pytorch)