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

谢谢。

解决方法

暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!

如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@)