RetinexFormer_arch网络结构导入问题

caiyuanhao1998 / Retinexformer

"Retinexformer: One-stage Retinex-based Transformer for Low-light Image Enhancement" (ICCV 2023) & (NTIRE 2024 Challenge)

https://arxiv.org/abs/2303.06705

MIT License

828 stars 64 forks source link

RetinexFormer_arch网络结构导入问题 #59

Closed thy0000 closed 5 months ago

thy0000 commented 5 months ago

作者大佬您好，我想请问一下，RetinexFormer_arch.py中的RetinexFormer网络结构是怎么导入到train.py中的呢？在model/init.py文件中，我看到代码是通过扫描model文件夹下后缀为_model.py的文件进行动态导入，但在base_model.py和image_restoration_model.py中，我好像没有看到从RetinexFormer_arch.py中导入RetinexFormer网络结构的函数，请问一下是怎么导入的呢？请您解答我的疑惑，谢谢！

caiyuanhao1998 commented 5 months ago

你好，感谢关注，在 basicsr/models/archs/__init__.py 里面会注册并导入以 __arch.py 结尾的网络结构文件