Pytorch 图像分割使用findContours报错

做图像分割时，想在Pytorch的预测结果的基础上进行后处理，用到了cv2.findContours里找标签的连通域，运行时报错：

cv2.error: OpenCV(4.0.0) /io/opencv/modules/imgproc/src/contours.cpp:195: error: (-210:Unsupported format or combination of formats) [Start]FindContours supports only CV_8UC1 images when mode != CV_RETR_FLOODFILL otherwise supports CV_32SC1 images only in function 'cvStartFindContours_Impl'

报错信息提示cv2.findContours的输入需要是uint8类型。第一个想法是直接修改numpy的dtype为np.uint8，结果标签的shape从512×512变成了512×4096，数据类型直接从pytorch转成numpy时类型是int64，直接修改dtype发现并没有对数据进行截断操作。

最后解决方案：在将tensor转换为numpy之前加上.byte()将tensor转换成uint8。

data_numpy = data.byte().cpu().detach().numpy()