数据处理小工具,xml和jpg相同文件名文件查找

提示：文章写完后，目录可以自动生成，如何生成可参考右边的帮助文档

---

前言

在使用labelimg标注结束数据后,生成的xml文件和imges中的jpg文件有时候会相差较多,如果数据集过大,导致我们在处理的时候会带来麻烦,为了能够使数据xml和jpg文件一一对应,用python进行了同文件名,不同文件夹不同文件类型的的查找.
提示：以下是本篇文章正文内容，下面案例可供参考

代码

代码如下：

#根据xml/json/txt文件找出对应的jpg文件
import os
import shutil

filepath1 = '/media/xx/2E1E25B31E2574CD/xxdata/data1/Anno'#源文件做参考
file_list = os.listdir(filepath1)
print(file_list)
print(len(file_list))

filepath2 = '/media/xx/2E1E25B31E2574CD/xxdata/data1/images/'#需要拷出的文件位置
filepath3 = '/media/xx/2E1E25B31E2574CD/xxmdata/data1/out/'#拷入新的文件夹
file_list1 = os.listdir(filepath2)
print(len(os.listdir(filepath2)))

def main():
    n=0
    for file in os.listdir(filepath1):
        aa,bb=file.split('.')
        b = aa +'.jpg'
        print("b",b)
        if b in file_list1:
            srcfile = filepath2 +'/'+ b
            dstfile = filepath3 +'/'+ b
            shutil.move(srcfile,dstfile)#剪切功能
            #shutil.copyfile(srcfile,dstfile)#拷贝出来
            n=n+1
            print(n)

if __name__ == '__main__':
    main()

总结

仅仅需要修改三个文件夹的路径,即可实现功能.

参考该博主:代码有点问题:https://blog.csdn.net/qq_44442727/article/details/112792866