【亲测免费】 Quick Draw 不规则遮罩数据集(QD-IMD) 使用与安装指南

本指南将带你深入了解如何安装并使用 Quick Draw Irregular Mask Dataset (QD-IMD)，一个基于5000万个手绘图案的开源数据集，旨在提供更为真实的图像修复任务训练数据。

1. 项目目录结构及介绍

QD-IMD项目的目录结构大致如下：

qd-imd/
├── generate_dataset.py         # 数据集生成脚本
├── README.md                   # 项目说明文档
├── requirements.txt            # 必需的Python包列表
├── .gitignore                 # 忽略的文件或文件夹配置
└── LICENSE                     # 许可证文件，遵循MIT协议

• generate_dataset.py: 负责根据设定参数生成不规则遮罩的数据集。
• requirements.txt: 包含项目运行所需的Python库列表。
• README.md: 提供项目概述、数据集生成方法和使用指导。
• .gitignore: 指定不需要纳入版本控制的文件类型或名称。
• LICENSE: MIT许可证文本，描述了代码的使用权限和限制。

2. 项目的启动文件介绍

generate_dataset.py

这是核心脚本，用于自动生成QD-IMD数据集。它的工作流程包括随机选取画笔笔触数量、从Quick Draw数据集中采样笔触、设置笔触宽度、进行画布放缩、中央裁剪到特定尺寸（默认512x512），以及二值化处理以形成最终的遮罩图。用户可以通过命令行参数定制生成过程。

3. 项目的配置文件介绍

QD-IMD项目并不直接提供一个传统意义上的配置文件，但其依赖于generate_dataset.py中的参数来定制数据生成的过程。这些参数并非存储在独立的配置文件中，而是通过脚本直接调用或者作为命令行参数来配置。例如，若需调整生成遮罩时使用的笔触平均数量、宽度范围或目标图片尺寸等，可通过修改脚本内部的相关变量或在执行脚本时提供相应的命令行选项来实现。

安装与使用步骤

步骤一：克隆项目

首先，通过Git克隆仓库到本地：

git clone https://github.com/karfly/qd-imd.git
cd qd-imd

步骤二：下载Quick Draw数据集简化版

你需要从Google云端硬盘获取Quick Draw数据集简化版。具体步骤见项目README.md。

步骤三：环境准备

确保安装所有必需的Python包：

pip install -r requirements.txt

步骤四：生成数据集

运行generate_dataset.py脚本来创建数据集。你可以查看帮助信息了解参数详情：

python generate_dataset.py --help

然后，按需求无参数运行或指定自定义参数来生成数据集：

python generate_dataset.py

通过以上步骤，你就成功地设置了QD-IMD数据集的开发环境，并可以根据需求生成自定义的不规则遮罩数据，进一步应用于图像修复或其他相关计算机视觉任务中。