A live demo of MIRNet-TFJS

78 views
Skip to first unread message

Rishit Dagli

unread,
Jan 28, 2021, 12:02:58 PM1/28/21
to TensorFlow.js Discussion
Hi,

A live demo of the MIRNet-TFJS repo, capable of enhancing low-light images to great extents is now out for you to try out.


More details in this tweet.

Es1Td4cUUAAdOwk.jfif

Jason Mayes

unread,
Jan 28, 2021, 2:32:50 PM1/28/21
to Rishit Dagli, TensorFlow.js Discussion
This is awesome work Rishit! Thank you for contributing to the community and making such epic models available in TFJS (even if via Node for now until the model size comes down for a client side deploy) :-)

If anyone has any thoughts on how to shrink this model for better browser consumption do reply :-)

Cheers!

Jason

--
You received this message because you are subscribed to the Google Groups "TensorFlow.js Discussion" group.
To unsubscribe from this group and stop receiving emails from it, send an email to tfjs+uns...@tensorflow.org.
To view this discussion on the web visit https://groups.google.com/a/tensorflow.org/d/msgid/tfjs/23f9aa83-5d9c-4ff8-8dfc-c318bce67dc6n%40tensorflow.org.


--




     
  •  Jason Mayes
  •  Senior Developer Advocate, TensorFlow.js
  •  Google
     Follow me or get inspired with #MadeWithTFJS demos

   

This email may be confidential and privileged. If you received this communication by mistake, please don't forward it to anyone else, please erase all copies and attachments, and please let me know that it has gone to the wrong person.


Sayak Paul

unread,
Feb 2, 2021, 6:46:39 AM2/2/21
to Jason Mayes, rishit...@gmail.com, TensorFlow.js Discussion
One immediate thing to be done is to compress the original model even further with distillation. 

This is on my radar since I worked on the TFLite variant of this same model: https://tfhub.dev/sayakpaul/lite-model/mirnet-fixed/dr/1

Some more context as to why distillation could be really helpful: in this notebook you will find 43 KB super resolution model distilled from a much larger model: 

Sayak Paul | sayak.dev

Jason Mayes

unread,
Feb 2, 2021, 4:25:32 PM2/2/21
to Sayak Paul, Rishit Dagli, TensorFlow.js Discussion
Wow that is quite the compression. Would love to see how fast this could potentially run in browsers with the optimizations you were using? Imagine using this in a browser to enhance low light webcam streams in real time. That would be epic.
Reply all
Reply to author
Forward
0 new messages