zengbohan0217 / CMVDM

AAAI 2024-Controllable Mind Visual Diffusion Model
12 stars 2 forks source link

ModuleNotFoundError: No module named 'src.midas' #1

Open xjh1213866 opened 7 months ago

xjh1213866 commented 7 months ago

Thanks very much for this wonderful work. I'm wondering where the src.midas and src.midas.midas_net_custom modules in this code, can you help me solve this question?

zengbohan0217 commented 7 months ago

Hello, thank you for your interest in our work. In fact, we did not use midas in this work, and you can choose to comment out the relevant code directly. Although the ideas proposed in this paper are indeed effective, there is still considerable room for optimization in silhouette extraction. We are still considering how to improve this aspect, and we welcome you to try optimizing this part.

NKUShaw commented 1 month ago

I run the stage A code with downloaded data and pretrains folder. However, I got this error "No such file or directory: 'data/Kamitani/npz/GOD_train_texts_tokens.pth' "