使用python OS和copy 将tf标记文件和源文件夹的对应图片的匹配和提出

将tf标记文件和源文件夹的对应图片的匹配和提出

书接上回，tf标记出来的数据集文件要和原始的文件匹配提取出来。
使用了OS和shutil库。
当初本来想用上re的，但是发现不需要那么多的功能。。。。
如果有需要就日后慢慢补充。

#!/usr/bin/env python
# coding: utf-8

import os
from shutil import copy

refer_list_dir = input('请输入参考文件夹的路径：')
from_list_dir = input('请输入源文件夹的路径：')
to_list_dir = input('移动到：')

os.chdir(from_list_dir)
print('当前工作路径为：{}'.format(from_list_dir))
for n in range(5):
    print('*',end = '\n')

refer_list = os.listdir(refer_list_dir)
from_list = os.listdir()

print('参考文件列表为：{}'.format(refer_list))
print('工作文件列表为：{}'.format(from_list))
for n in range(5):
    print('*',end = '\n')

i = 0
a = 0
b = 0
for obj in refer_list:
    new_obj = obj[:-4] + '.txt'
    from_path = os.path.join(from_list_dir,new_obj)
    to_path = os.path.join(to_list_dir,new_obj)
    try:
        copy(from_path,to_path)
        print('将{}复制到{}'.format(from_path,to_path))
        print('*****************************************')
        a += 1
    except:
        print('复制出错，请手动检查\n{}\n和\n{}\n两处文件是否存在和匹配'.format(from_path,os.path.join(refer_list_dir,obj)))
        print('*****************************************')
        b += 1
    i += 1
print('任务完成，共计操作{}个项目，成功{}个，失败{}个'.format(i,a,b))

input('按下回车键后退出')