FiftyOne——指定类别下载CoCo和Open-images数据集

客官°小女子只卖身不卖艺 2022-09-02 12:46 865阅读 0赞

### 文章目录 ###

*  FiftyOne——指定类别下载CoCo和Open-images数据集
 *   *  引言
     *  指定类别下载COCO数据
     *  导入已下载的COCO数据集
     *  指定类别下载谷歌Open-image-V6
     *  问题记录
     *   *  目标计算机积极拒绝
     *  附录
     *   *  1 目前支持的公开数据集
         *  2 foz.load\_zoo\_dataset()函数的详细解释

# FiftyOne——指定类别下载CoCo和Open-images数据集 #

## 引言 ##

[官方文档][Link 1]  
[安装与简单测试见另外一篇博客][Link 2]  
FiftyOne可以指定条件下载CoCo等大型数据集的部分数据集。

系统版本:Win10专业版  
Python版本:3.7

\*\*

> 使用FiftyOne需要理清以下两个概念：  
> 数据：指的是具体的图像数据、如COCO数据 数据集  
> dataset：指的是一个数据的组织关系。里面组织了指向的数据，以及数据相应的信息。
> 
> 就是说数据集实际上就是一个组织关系，例如你1个文件夹中有100张图片的数据，可以创建N个数据集，例如随机取其中的20张做一个数据集，随机30张做一个数据集…  
> 这里的数据集指的是一个抽象的组织关系，不是具体的数据。 FiftOne在创建数据集的时候，不会对原始图像数据进行拷贝操作。  
> \*\*

## 指定类别下载COCO数据 ##

直接把解释放在代码里面。  
如果是一次下载coco数据集的话，即便指定的样本数量较少，也需要下载将近1个G的内容，这是因为完整的注视文件就将近1个G。尽量不要随意指定数据集保存的路径，如果更换的话，建议将原本文件夹中的注释文件一并复制到新的路径下。

import fiftyone as fo
    import fiftyone.zoo as foz
    
    if __name__ == "__main__":  # Win10情况下必须把foz.load_zoo_dataset放到main下，否则多进程将会报错
    
        dataset = foz.load_zoo_dataset(  # 想要看输入参数的解释的话，选择load_zoo_dataset 按 ctrl+q（pycharm IDE中）
            "coco-2017",  # 指定下载coco-2017类型
            split="validation",  # 指定下载验证集
            label_types=["detections"],  # 指定下载目标检测的类型
            classes=["cat"],  # 指定下载猫的类别
            max_samples=10,  # 指定下载符合条件的最大样本数
            only_matching=True,  # 指定仅下载符合条件的图片，即含有猫的图片
            num_workers=1,  # 指定进程数为1
            dataset_dir="J:\\A1COCO",  # 指定下载的数据集保存的路径,尽量不要随意更改，这是保存原始图片的路径，
            dataset_name="open-c",  # 指定新下载的数据集的名称,会检测是否已有,不同的dataset 都是指向了同一个原始图像的路径
        )
        session = fo.launch_app(dataset)
        session.wait()  #

下载后的文件夹目录如下，主要有2个文件夹和一个json文件。 其中raw文件夹中存放的是COCO数据集原始的注释文件，即官方文件。 Validation文件夹即本次下载的数据集数据。info文件中存放的是最新的一次下载的信息。  
![在这里插入图片描述][4c0f59cc7b3d4590a4f7eb46baf58204.png]  
Validation中包含了一个data文件夹和一个labels.json文件，这是FiftyOne的固定格式。其中，data文件夹中存放的是图片，labels是对应的注释文件。  
![在这里插入图片描述][23e83b0e9c324d9c9cda72b5365f3dfb.png]  
**注意：dataset建立与图像数据集下载是两个概念， 如果要建立的dataset的图像已经被下载到当前的文件夹中了，将不会重新下载。反则下载。 就是说，如果我有100张图片，我在Fiftyone中可以取前30张建立dataset A，也可以取后30张建立dataset B，这个不会对原始的100张图片进行拷贝，相当于只是建立了索引信息。但是当我想要取150张图片建立dataset C的时候，因为现在的文件夹中符合条件的只有100张，那么他将会再下载50张下来。这种做法可以很好的节省了硬盘的空间**

代码中的dataset\_name名字并不会随着程序的结束而自动清零，他会被离线保存，但是不会对原始的图片进行拷贝，大概的意思是保存的是数据集的信息和图片的索引。因此在新建数据集的时候，应该看一下现有的dataset有哪些，会不会冲突。

网页中的效果如下图，其中包含了车等其它物品的标签。  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3NjeTI2MTk4MzYyNg_size_16_color_FFFFFF_t_70]  
当指定仅下载猫这个类型的标签的时候，如下图  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3NjeTI2MTk4MzYyNg_size_16_color_FFFFFF_t_70 1]

## 导入已下载的COCO数据集 ##

################################################
    # 如果本身已经下载好了coco数据集，那么采用下面的方式来导入
    ################################################
    pth_标签文件路径 = "J:\\A1COCO\\raw\\instances_val2017.json"  # 这边的标签文件是用的前面下载的。
    pth_图片文件夹路径 = "J:\\COCO\\val2017"  # 这个文件夹中是我之前下载的coco数据集中val系列的所有照片
    
    # Import the dataset
    dataset = fo.Dataset.from_dir(
        dataset_type=fo.types.COCODetectionDataset,
        data_path=pth_图片文件夹路径,
        labels_path=pth_标签文件路径,
        classes=["cat", "apple"],  # 指定载入猫的类别
        only_matching=True,  # 指定仅下载符合条件的图片，即含有猫的图片
    )
    dataset.name = "coco2"
    session = fo.launch_app(dataset)
    session.wait()

载入效果  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3NjeTI2MTk4MzYyNg_size_16_color_FFFFFF_t_70 2]

> 创建新的dataset的时候使用 fo.Dataset();载入已有的dataset的时候使用fo.load\_dataset()这样可以更省时间。 查看现有dataset可以用fo.list\_datasets()

## 指定类别下载谷歌Open-image-V6 ##

谷歌Open-image-v6是由谷歌出资标注的一个超大型数据集，数据大小达到600多G，类别达到600多种分类，对于普通研究者而言，根本没办法全部下载下来做测试，也没必要。只需要下载与自己任务相关的数据集即可。

*  我们先随便指定一个类别，比如苹果，设置最大下载样本数为1，可以先拿到包含所有类别的文件。

if __name__ == '__main__':
        dataset_test = foz.load_zoo_dataset(
            "open-images-v6",
            split="test",
            classes=["Apple"],
            max_samples=1,
            shuffle=True,
            only_matching=True,  
            label_types=["detections"],  # 指定下载目标检测的类型,detections,
            dataset_dir="J:\\A2OpenImageV6",# 保存的路径
            num_workers=2,  # 指定工作进程数
        )

*  运行后，会先下载注释文件和相关信息文件  
    ![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3NjeTI2MTk4MzYyNg_size_16_color_FFFFFF_t_70 3]
 *  下载后的文件中含有一个classes文件，里面包含了600多个类别，然后在里面找到自己感兴趣的类别，进一步下载就可以了

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3NjeTI2MTk4MzYyNg_size_16_color_FFFFFF_t_70 4]

## 问题记录 ##

### 目标计算机积极拒绝 ###

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3NjeTI2MTk4MzYyNg_size_16_color_FFFFFF_t_70 5]  
原因：使用了VPN代理  
解决方案：在IE中取消代理

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3NjeTI2MTk4MzYyNg_size_16_color_FFFFFF_t_70 6]![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3NjeTI2MTk4MzYyNg_size_16_color_FFFFFF_t_70 7]

## 附录 ##

### 1 目前支持的公开数据集 ###

[官方文档][Link 3]  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3NjeTI2MTk4MzYyNg_size_16_color_FFFFFF_t_70 8]

### 2 foz.load\_zoo\_dataset()函数的详细解释 ###

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3NjeTI2MTk4MzYyNg_size_16_color_FFFFFF_t_70 9]

[Link 1]: https://voxel51.com/
[Link 2]: https://blog.csdn.net/scy261983626/article/details/119327350?spm=1001.2014.3001.5501
[4c0f59cc7b3d4590a4f7eb46baf58204.png]: /images/20220829/94913d6b56a34ac0a268a8d4b5e0eb3c.png
[23e83b0e9c324d9c9cda72b5365f3dfb.png]: /images/20220829/df02bad049f34077be2943fa44be3fff.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3NjeTI2MTk4MzYyNg_size_16_color_FFFFFF_t_70]: /images/20220829/95a722411449454f8d77f8396a9b7400.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3NjeTI2MTk4MzYyNg_size_16_color_FFFFFF_t_70 1]: /images/20220829/6e20f8dc036e40fcb22aeba6e20b77f5.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3NjeTI2MTk4MzYyNg_size_16_color_FFFFFF_t_70 2]: /images/20220829/154d24dc3ea049b1a08d2c563ffd4aee.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3NjeTI2MTk4MzYyNg_size_16_color_FFFFFF_t_70 3]: /images/20220829/71fac84efceb4598b5ef5e22cad8ba3e.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3NjeTI2MTk4MzYyNg_size_16_color_FFFFFF_t_70 4]: /images/20220829/c494fdb932a64a84aba24f553bc9ea9c.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3NjeTI2MTk4MzYyNg_size_16_color_FFFFFF_t_70 5]: /images/20220829/a6bcf741fdd240da88dfd62193083fa9.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3NjeTI2MTk4MzYyNg_size_16_color_FFFFFF_t_70 6]: https://img-blog.csdnimg.cn/8b72926768b34771bd3cd26c9e5dd538.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3NjeTI2MTk4MzYyNg==,size_16,color_FFFFFF,t_70
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3NjeTI2MTk4MzYyNg_size_16_color_FFFFFF_t_70 7]: /images/20220829/1f59fa481a1e451f9f00e5cbbc241b66.png
[Link 3]: https://voxel51.com/docs/fiftyone/user_guide/dataset_zoo/datasets.html#dataset-zoo-datasets
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3NjeTI2MTk4MzYyNg_size_16_color_FFFFFF_t_70 8]: /images/20220829/05e8ae3b1654483aa4d467d0f6cd893e.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3NjeTI2MTk4MzYyNg_size_16_color_FFFFFF_t_70 9]: /images/20220829/9aa7ff3709324f6cb2133d583e1d3f71.png