labelme_to_json_dataset 报错

我用labelme给YaleB数据库的人脸图片做分割标签,在使_用labelme____to_json_dataset_时报错,提示信息如下:_AssertionError: rgb must be 3 dimensional_

我也遇到了这个问题,可以试着把图片的格式改成.jpg,我是把格式从.png改成.jpg成功的;
png格式其实是RGBA模式,jpg是RGB模式,源码中有一处断言判断图片对应矩阵的shape的,assert rgb.shape[2] == 3, 'rgb shape must be (H, W, 3),所以.png格式的图片会报错