search

Rishit-dagli / MIRNet-TFJS

TensorFlow JS models for MIRNet for low-light💡 image enhancement
https://rishit-dagli.github.io/MIRNet-TFJS/
Apache License 2.0
350 stars 38 forks source link

Any idea on inference with TFJS? #2

Closed pottekkat closed 3 years ago

pottekkat commented 3 years ago

The code looks good and the converted models look good too. But, how do we perform inference with TFJS? Seems like you have performed inference, not on the TFJS models and it uses a function from mirnet. So I am guessing we have to implement something similar for TFJS.

If anyone has any idea or implementation for that it would be really helpful!

@Rishit-dagli Great work!

pottekkat commented 3 years ago

NVM I think it does not do much, just reshape the image and convert it into array.

I should have checked the code before!

Rishit-dagli commented 3 years ago

@navendu-pottekkat Thank You! I had that in the pipeline and planned to get out an example after the model is published on TF Hub. However, it makes a lot of sense to release a live demo and I have started working on that.

Thanks a lot for the suggestion.

Rishit-dagli commented 3 years ago

Hey @navendu-pottekkat , the implementation part have now been pushed an the details about the same have also been updates in the README. Let me know if you have any more questions.

pottekkat commented 3 years ago

@Rishit-dagli Great! I will take a look