问题描述
因此,我尝试按照 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 (将#修改为@)