search

htyjers / StrDiffusion

[CVPR 2024] Structure Matters: Tackling the Semantic Discrepancy in Diffusion Models for Image Inpainting
Apache License 2.0
85 stars 11 forks source link

有关输入数据通道数的问题 #22

Closed vfcerexwn closed 2 weeks ago

vfcerexwn commented 4 months ago

您好，请问如果想处理灰度图，代码需要在哪里作出修改，谢谢！

htyjers commented 4 months ago

我没太明白您的意思，你是想在哪个分支处理灰度图，structure分支本身修复结果就是灰度图，texture分支是修复RGB图像的，您可以给出一些细节信息吗

paco7er commented 2 months ago

如果我想直接传入灰度图，该如何修改代码， File "/autodl-fs/data/StrDiffusion/test/texture/config/inpainting/data/GT_dataset.py", line 128, in getitem if img_GT.shape[2] == 3: IndexError: tuple index out of range 现在出现在了这个问题