IBM开发人员模型资产交换:图像分割
该存储库包含用于实例化和部署图像分割模型的代码。 该模型将图像文件作为输入,并返回包含输入图像中每个像素的预测类的分割图。
该存储库包含2个在PASCAL VOC 2012上训练的模型。一个模型是使用xception架构进行训练的,可以产生非常准确的结果,但是要花几秒钟的时间运行,而另一个模型在MobileNetV2上进行训练,虽然速度更快但准确性较低。 您可以在启动Docker映像时指定希望使用的模型。 请参阅下面的更多细节。
分割图针对输入图像中的每个像素返回0到20之间的整数,该整数与下面的标签之一相对应。 第一个嵌套数组对应于图像中像素的顶行,而该数组中的第一个元素对应于图像左上角的像素。 注意:图像将被调整大小,并且分割图是指调整后的图像中的像素,而不是原始输入图像。
ID
标签
ID
标签
ID
标签
0
背景
7
车
14
评论0
最新资源