infer.py中PIL转opencv不显示
Created by: xmy0916
我把infer.py里面预测结果保存的代码做了点修改想用opencv显示,修改如下:
# 预测结果保存
result_path = os.path.join(cfg.vis_dir, im_name + '.png')
output_im = PILImage.fromarray(np.asarray(parsing, dtype=np.uint8))
output_im.putpalette(palette)
output_im.show()
img = cv2.cvtColor(np.asarray(output_im), cv2.COLOR_RGB2BGR) # PIL转cv
cv2.imshow("cvPicture",img)
cv2.waitKey(0)
output_im.save(result_path)
我将最后要写入本地的PIL图片转成了opencv的mat图片,但是imshow出来是一片黑的,请问您知道为什么嘛,需要怎么修改呢?显示结果如下: