albrateanu / LYT-Net

LYT-Net: Lightweight YUV Transformer-based Network for Low-Light Image Enhancement
https://arxiv.org/abs/2401.15204
MIT License
70 stars 12 forks source link

Do you consider migrating Python versions? #10

Closed stuhao251 closed 4 months ago

stuhao251 commented 6 months ago

Hello, I am curious if you will release a PyTorch version of the project. The current TF version is not suitable for operation and learning

albrateanu commented 6 months ago

Hi, thanks for writing.

We will provide a PyTorch version to facilitate better understanding. Due to other projects going on, we don't estimate a faster release than June-July.

albrateanu commented 4 months ago

Hello stuhao251, PyTorch version has been added, detailed environment and code setup guide coming in the following days.

stuhao251 commented 4 months ago

Thank you for your contribution!