search

mdyao / HDR-BiTNet

[TMM 2023] Official Implementation of "Bidirectional Translation Between UHD-HDR and HD-SDR Videos"
MIT License
5 stars 0 forks source link

How to run a simple png -> png with this code? #2

Open GeneralShan opened 5 months ago

GeneralShan commented 5 months ago

How can I use this code to run a simple PNG to PNG image test? I see your example code but it does not show how you have made the very impressive video example? I would like to learn how to do this.

GeneralShan commented 5 months ago

I see there are some codes to run the HDR -> SDR. But I want to know how I can input SDR image and output a HDR image using this project.

mdyao commented 5 months ago

Hi @GeneralShan , you can simply change the dataloader to suit the PNG input/output. I would like to add the corresponding codes later. If there are any obstacles, feel free to contact me.

I use the ffmepg tool to make a video example.

You can adjust the rev=True/False parameter in self.netG(x=y_, rev=True) to convert between HDR and SDR, as needed.

mdyao commented 5 months ago

@GeneralShan I will refactor the code later to meet all you mentioned requirements, no later than the end of this month.