search

zsyOAOA / ResShift

ResShift: Efficient Diffusion Model for Image Super-resolution by Residual Shifting (NeurIPS@2023 Spotlight, TPAMI@2024)
Other
944 stars 50 forks source link

有关三维化的问题 #77

Open selfish0618 opened 4 months ago

selfish0618 commented 4 months ago

作者您好，感谢您的工作。 我想对您的网络进行三维化处理，使其可以在医学图像上进行工作。 但是在 preparing stage 中需要“Download the pre-trained VQGAN model from this [link]”。这是否意味着我需要先去预训练一个三维化的VQGAN才能将其应用到我想要三维化的网络中。 感谢您的回复和建议。

Hiilay commented 3 months ago

您好，是否已经有答案了，能麻烦告知一下吗谢谢！

zsyOAOA commented 3 months ago

是的，你需要针对医学图像重新训练一个VQGAN，或者直接在pixel space上做也可以。

Hiilay commented 3 months ago

@zsyOAOA 感谢回复！另外想问一下如何重新训练一个三维的VQGAN呢，您是否有一些推荐？盼复！

zsyOAOA commented 3 months ago

建议试用VQGAN的官方代码去训练。