maskrcnn_benchmark中COCODataset类的图片类型

如何解决maskrcnn_benchmark中COCODataset类的图片类型

因此，我尝试按照 this 指令使用 maskrcnn_benchmark 来可视化我的测试结果。由于图像类型不同，我无法输出图像。对于 image,target,idx = dataset[idx]，我从数据集 torch.Tensor 中获得了一个数据类型为 maskrcnn_benchmark.data.datasets.coco.COCODataset 的图像，形状为 (3,600,1200)。

来自 dataset.get_img_info(idx) 的原始图像是 {'id': 0,'width': 2048,'height': 1024,'file_name': 'frankfurt_000000_000294_leftImg8bit_foggy_beta_0.02.png','seg_file_name': 'frankfurt_000000_000294_gtFine_instanceIds.png'}

问题是 torch.Tensor 图像值在 -122.7717 到 152.0199 之间变化。我想知道如何将其转换为普通图像类型，以便我可以将预测叠加在图像之上并可视化测试结果。在将预测叠加到图像之上之前，我是否还需要将图像重塑为原始大小 1024 x 2048？

谢谢。