图片数据与对应标签数据互相调出

• 通过图片文件名调出对应的标签数据

import glob
import cv2
import os
import shutil



if __name__ == '__main__':
    imgPath = 'E:\\original-data\\picture\\test\\img'
    savePath = 'E:\\original-data\\picture\\test\\save\\'
    filename = os.listdir(imgPath)
    totalTxt = "E:\\original-data\\picture\\label\\"
    for fn in filename:
        fn = fn.replace("png", "txt")
        txtPath = totalTxt + fn
        shutil.copyfile(txtPath, savePath + fn)

• 通过标签文件名调出对应的图片数据

import glob
import cv2
import os


if __name__ == '__main__':
    txtPath = 'E:\\original-data\\picture\\test\\txt'
    savePath = 'E:\\original-data\\picture\\test\\save\\'
    filename = os.listdir(txtPath)
    totalImg = "E:\\original-data\\picture\\pic\\"
    for fn in filename:
        fn = fn.replace("txt", "png")
        ImgPath = totalImg + fn
        img = cv2.imread(ImgPath)
        cv2.imwrite(savePath + fn, img)