Group Abstract Group Abstract

Message Boards Message Boards

Deep Learning Library for Mathematica

Posted 9 years ago

Dear all,

I have been working on an open source deep learning library in Mathematica which I would like to share.

The code is hosted on github under an MIT license: https://github.com/jfrancis71/Machine-Vision

I have succesfully used it to train a face recognition system, and also automatically drive my lego EV3 robot.

Face Recognition: https://www.youtube.com/watch?v=TdRtUnSppB0

EV3 Car: https://www.youtube.com/watch?v=DCad82UdDFA

This has arisen from a homegrown project, so the code is probably a little scruffy (and not well documented) at the moment. Having said that it does work and I would welcome constructive feedback and help turning this into an active open source project.

It's not really aimed at someone who just wants a good face recogniser, the new built in Mathematica functions are probably more suitable for that. it's more suitable for someone interested in the internals or developing their own neural nets.

The distinction with the current Mathematica machine learning functionality, is this library is designed to support multiple layer networks including convolutional layers (much used in computer vision presently).

There is more information available on the github site in the link above.

Thanks, Julian Francis.

POSTED BY: Julian Francis
7 Replies
Posted 9 years ago

Julian, I am working in a very similar field and have been building generalized machine learning engines using Mathematica 10. Shoot me an e-mail if you would like to chat about your work and it's direction. I believe there could be some heavy overlap in what we are doing and I would like to share what I have been doing with you. My e-mail is wduhe@yottaforce.co and my Skype ID is william.john.pierre.duhe - would love to hear back from you soon.

POSTED BY: William Duhe

I was following this GPU acceleration tutorial that uses OpenCLLink and CUDALink with a package for RecurrentNeuralNetworks but the link to download the RNN package is broken. I reached out to the developer and his email is bouncing.

http://www.ims2015.net/proceedings/papers/29.pdf

I think this will be very helpful for you though. I have a beefy NVIDIA card and would love to get something working.

BTW: I have a small budget to contribute if you are interested. :)

POSTED BY: David Johnston
Posted 9 years ago
POSTED BY: Julian Francis
Posted 9 years ago
POSTED BY: Julian Francis

Great work on this. You obviously spent a lot of time on the Wiki and all that. Looks great.

I really appreciate your hard work and willingness to share it.

Are you thinking of building a similar package for RNN's? I would love to do cool stuff like DeepDream does.

POSTED BY: David Johnston
Posted 9 years ago

Thanks for sharing Julian :)

POSTED BY: Thomas Eli

@Julian Francis, this is a very nice project, thanks for sharing! You might consider applying for Wolfram Summer School where similar projects are worked on. You would have a chance to connect with Wolfram folks who work on built-in machine learning tools and also with many international students with similar interests.

POSTED BY: Vitaliy Kaurov
Reply to this discussion
Community posts can be styled and formatted using the Markdown syntax.
Reply Preview
Attachments
Remove
or Discard