【deeplab数据集3】批量提取label图像

r囧r小猫 2021-06-11 15:14 461阅读 0赞

**目录**

前言

一、批量提取

二、图片重命名

--------------------

# 前言 #

最近在接触一个项目，要用到语义分割，如果大家有什么问题，都可以跟我沟通交流。除了通过博客交流外，欢迎你加入我的QQ群：**326866692**，一起交流有关于区块链、机器学习、深度学习、计算机视觉有关内容。我现在处于广泛涉猎阶段，希望我们能够一起沟通。下图是我的群二维码：

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3NodWl5aXhpbg_size_36_color_F369FF_t_70][]

上一次我们说到了使用labelme将json文件转化为数据集，转化完成后，每一张图片都生成了一个单独的文件夹，如果我们一个一个把里面的label图像取出来很麻烦，特别是数据量极大地时候，比如我们经常会做好几万的数据集，这个时候再手动取出来就是没事闲的了。

所以我们用代码，将图片集体取出。

# 一、批量提取 #

先创建一个python文件，并写入以下代码：

import os
    import random
    import shutil
    import re
    
    GT_from_PATH = "D:/zyt/JSONALL/jsonNew"
    GT_to_PATH = "D:/zyt/JSON/ALL1"
    
    
    def copy_file(from_dir, to_dir, Name_list):
        if not os.path.isdir(to_dir):
            os.mkdir(to_dir)
        # 1
        # name_list = os.listdir(from_dir)
    
        # # 2
        # sample = random.sample(pathDir, 2)
        # print(sample)
    
        # 3
        for name in Name_list:
            try:
                # print(name)
                if not os.path.isfile(os.path.join(from_dir, name)):
                    print("{} is not existed".format(os.path.join(from_dir, name)))
                shutil.copy(os.path.join(from_dir, name), os.path.join(to_dir, name))
                # print("{} has copied to {}".format(os.path.join(from_dir, name), os.path.join(to_dir, name)))
            except:
                # print("failed to move {}".format(from_dir + name))
                pass
            # shutil.copyfile(fileDir+name, tarDir+name)
        print("{} has copied to {}".format(from_dir, to_dir))
    
    if __name__ == '__main__':
        filepath_list = os.listdir(GT_from_PATH)
        # print(name_list)
        for i, file_path in enumerate(filepath_list):
            gt_path = "{}/{}_gt.png".format(os.path.join(GT_from_PATH, filepath_list[i]), file_path[:-5])
            print("copy {} to ...".format(gt_path))
            gt_name = ["{}_gt.png".format(file_path[:-5])]
            gt_file_path = os.path.join(GT_from_PATH, file_path)
            copy_file(gt_file_path, GT_to_PATH, gt_name)

其中，我们需要修改的有如下两个内容：

GT_from_PATH = "" #json生成的文件夹的父文件夹
    
    GT_to_PATH = ""  #label输出路径

然后执行代码即可。执行完后的结果如下：

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3NodWl5aXhpbg_size_36_color_F7536F_t_70][]

# 二、图片重命名 #

可以使用代码，也可以使用软件，我使用的是2345看图王，可以批量修改图片名称，重命名的目的是将图片后面的 “**\_gt**” 去掉。

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3NodWl5aXhpbg_size_36_color_F7256F_t_70][]

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3NodWl5aXhpbg_size_36_color_FFF41F_t_70][]

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3NodWl5aXhpbg_size_36_color_F369FF_t_70]: /images/20210516/bcc3489651384aad8105444fe9721f8c.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3NodWl5aXhpbg_size_36_color_F7536F_t_70]: /images/20210516/c3d792f20b404c409a68d954d12e092f.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3NodWl5aXhpbg_size_36_color_F7256F_t_70]: /images/20210516/85334bb51b324e12bf6fc4c8c14ed07c.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3NodWl5aXhpbg_size_36_color_FFF41F_t_70]: /images/20210516/75aae9362fd74384b55ca3feab44ca3a.png