Supervisely 数据集的读取和解析有两种方法:1. 读取 json 文件自己解析;2. 使用 supervisely_lib 解析。
人像分割数据集中人像的标注有两种几何类型:位图(bitmap),多边形(polygon)。自己解析需要对这两种类型进行处理,参考 “Supervisely 人像分割数据集格式解析”。个人建议使用方法2。supervisely_lib 提供了统一的API,你不需要关心几何类型,它们(位图,多边形)都可以使用相同的API获取 mask,轮廓,box。
这里以生成 Mask 图像为例。
一、安装 supervisely_lib
supervisely_lib 项目地址:https://github.com/supervisely/supervisely
安装方法参见项目主页(README.md)。
安装过程中会依赖一些第三方库,其中要求 opencv-python 版本小于4.0.0。因为我的Python环境中已安装 opencv-python 4.2.0,所以我对 supervisely 稍微的修改,差异如下:
diff --git a/setup.py b/setup.py
index dad24a3..4cc0698 100644
--- a/setup.py
+++ b/setup.py
@@