The hdr-bitnet from mdyao

mdyao / hdr-bitnet Goto Github PK

View Code? Open in Web Editor NEW

[TMM 2023] Official Implementation of "Bidirectional Translation Between UHD-HDR and HD-SDR Videos"

License: MIT License

Python 100.00%

hdr-bitnet's People

Contributors

Stargazers

Watchers

hdr-bitnet's Issues

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

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.

Not able to run the test

Hello, thank you for your repository!

I am having trouble executing the test, I was hoping you could provide some guidance @mdyao ?

I got as far as modifying this line to load the SDR file but I then get the following error below. Do you have any advice what I am doing wrong?

Original Traceback (most recent call last):
  File "C:\Python39\lib\site-packages\torch\utils\data\_utils\worker.py", line 308, in _worker_loop
    data = fetcher.fetch(index)
  File "C:\Python39\lib\site-packages\torch\utils\data\_utils\fetch.py", line 51, in fetch
    data = [self.dataset[idx] for idx in possibly_batched_index]
  File "C:\Python39\lib\site-packages\torch\utils\data\_utils\fetch.py", line 51, in <listcomp>
    data = [self.dataset[idx] for idx in possibly_batched_index]
  File "C:\Users\Development\HDR-BiTNet\data\JSH_dataset_val.py", line 42, in __getitem__
    S = extrac_structure(HDR_img, SDR_img)
  File "C:\Users\Development\HDR-BiTNet\models\extrac_S.py", line 50, in extrac_structure
    out, HDR, SDR = extract_S(hdr, sdr)
  File "C:\Users\Development\HDR-BiTNet\models\extrac_S.py", line 36, in extract_S
    out = (SDR + HDR)/2
ValueError: operands could not be broadcast together with shapes (1,1,3840,2160) (1,1,1920,1080)

Recommend Projects