Yaziwel / All-In-One-Medical-Image-Restoration-via-Task-Adaptive-Routing

[MICCAI 2024] All-In-One Medical Image Restoration via Task-Adaptive Routing (AMIR).
Apache License 2.0
35 stars 2 forks source link

数据集的问题 #3

Closed gejutaixiao22 closed 4 months ago

gejutaixiao22 commented 4 months ago

你好我想询问一下CT去噪中那个训练集是.bin文件,我该如何把它转为.png的文件。

Yaziwel commented 4 months ago

你好我想询问一下CT去噪中那个训练集是.bin文件,我该如何把它转为.png的文件。

bin文件存储的是tensor, 可以通过以下代码对bin进行读取:

from .common import dataIO
io = dataIO()
img_tensor = io.load("***.bin") #此处得到的是tensor

img_tensor 是训练CT的patch,数值范围是[-1024, 3071], 你可以进行缩放以及取整操作变成[0, 255]范围