我正在try 将用pytorch_mobile包预测的二进制掩码转换为可以在我的应用程序中显示的图像.
我收到的预测是一个一维列表,其中包含我的模型输出的预测,这些预测对于分配给背景的像素是负值,对于分配给感兴趣区域的像素是正值.在此之后,我创建了一个列表,将值0赋给所有先前的负值,将255赋给所有先前的正值,从而生成一个一维列表,该列表包含值0或255,具体取决于像素的分类.
图像预测是512×512像素的大小,并且列表的长度随后是262,144.
我如何才能将此列表转换为可以保存到存储或通过Ffltter用户界面显示的图像?
以下是我当前的代码:
customModel = await PyTorchMobile
.loadModel('assets/segmentation_model.pt');
result_list = [];
File image = File(filePath);
List prediction = await customModel.getImagePredictionList(image, 512, 512);
prediction.forEach((element) {
if (element >0){
result_list.add(255);
}else if(element <= 0){
result_list.add(0);
}
});
result_list_Uint8 = Uint8List.fromList(result_list);